Skip to content

Commit 8174771

Browse files
avichaymAvichay Marciano
andauthored
[api][integration] Add ResourceName constants for AWS integrations (#679)
Co-authored-by: Avichay Marciano <avichaym@github.com>
1 parent b38ae21 commit 8174771

2 files changed

Lines changed: 34 additions & 0 deletions

File tree

api/src/main/java/org/apache/flink/agents/api/resource/ResourceName.java

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -59,6 +59,12 @@ public static final class ChatModel {
5959
public static final String AZURE_SETUP =
6060
"org.apache.flink.agents.integrations.chatmodels.azureai.AzureAIChatModelSetup";
6161

62+
// Bedrock
63+
public static final String BEDROCK_CONNECTION =
64+
"org.apache.flink.agents.integrations.chatmodels.bedrock.BedrockChatModelConnection";
65+
public static final String BEDROCK_SETUP =
66+
"org.apache.flink.agents.integrations.chatmodels.bedrock.BedrockChatModelSetup";
67+
6268
// Ollama
6369
public static final String OLLAMA_CONNECTION =
6470
"org.apache.flink.agents.integrations.chatmodels.ollama.OllamaChatModelConnection";
@@ -131,6 +137,12 @@ public static final class EmbeddingModel {
131137
public static final String OLLAMA_SETUP =
132138
"org.apache.flink.agents.integrations.embeddingmodels.ollama.OllamaEmbeddingModelSetup";
133139

140+
// Bedrock
141+
public static final String BEDROCK_CONNECTION =
142+
"org.apache.flink.agents.integrations.embeddingmodels.bedrock.BedrockEmbeddingModelConnection";
143+
public static final String BEDROCK_SETUP =
144+
"org.apache.flink.agents.integrations.embeddingmodels.bedrock.BedrockEmbeddingModelSetup";
145+
134146
// Python Wrapper
135147
public static final String PYTHON_WRAPPER_CONNECTION =
136148
"org.apache.flink.agents.api.embedding.model.python.PythonEmbeddingModelConnection";
@@ -171,6 +183,14 @@ public static final class VectorStore {
171183
public static final String ELASTICSEARCH_VECTOR_STORE =
172184
"org.apache.flink.agents.integrations.vectorstores.elasticsearch.ElasticsearchVectorStore";
173185

186+
// Amazon OpenSearch (Serverless or Service domains)
187+
public static final String OPENSEARCH_VECTOR_STORE =
188+
"org.apache.flink.agents.integrations.vectorstores.opensearch.OpenSearchVectorStore";
189+
190+
// Amazon S3 Vectors
191+
public static final String S3_VECTORS_VECTOR_STORE =
192+
"org.apache.flink.agents.integrations.vectorstores.s3vectors.S3VectorsVectorStore";
193+
174194
// Python Wrapper
175195
public static final String PYTHON_WRAPPER_VECTOR_STORE =
176196
"org.apache.flink.agents.api.vectorstores.python.PythonVectorStore";

python/flink_agents/api/resource.py

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -276,6 +276,10 @@ class Java:
276276
AZURE_CONNECTION = "org.apache.flink.agents.integrations.chatmodels.azureai.AzureAIChatModelConnection"
277277
AZURE_SETUP = "org.apache.flink.agents.integrations.chatmodels.azureai.AzureAIChatModelSetup"
278278

279+
# Bedrock
280+
BEDROCK_CONNECTION = "org.apache.flink.agents.integrations.chatmodels.bedrock.BedrockChatModelConnection"
281+
BEDROCK_SETUP = "org.apache.flink.agents.integrations.chatmodels.bedrock.BedrockChatModelSetup"
282+
279283
# Ollama
280284
OLLAMA_CONNECTION = "org.apache.flink.agents.integrations.chatmodels.ollama.OllamaChatModelConnection"
281285
OLLAMA_SETUP = "org.apache.flink.agents.integrations.chatmodels.ollama.OllamaChatModelSetup"
@@ -313,6 +317,10 @@ class Java:
313317
OLLAMA_CONNECTION = "org.apache.flink.agents.integrations.embeddingmodels.ollama.OllamaEmbeddingModelConnection"
314318
OLLAMA_SETUP = "org.apache.flink.agents.integrations.embeddingmodels.ollama.OllamaEmbeddingModelSetup"
315319

320+
# Bedrock
321+
BEDROCK_CONNECTION = "org.apache.flink.agents.integrations.embeddingmodels.bedrock.BedrockEmbeddingModelConnection"
322+
BEDROCK_SETUP = "org.apache.flink.agents.integrations.embeddingmodels.bedrock.BedrockEmbeddingModelSetup"
323+
316324
class VectorStore:
317325
"""VectorStore resource names."""
318326

@@ -331,5 +339,11 @@ class Java:
331339
# Elasticsearch
332340
ELASTICSEARCH_VECTOR_STORE = "org.apache.flink.agents.integrations.vectorstores.elasticsearch.ElasticsearchVectorStore"
333341

342+
# Amazon OpenSearch (Serverless or Service domains)
343+
OPENSEARCH_VECTOR_STORE = "org.apache.flink.agents.integrations.vectorstores.opensearch.OpenSearchVectorStore"
344+
345+
# Amazon S3 Vectors
346+
S3_VECTORS_VECTOR_STORE = "org.apache.flink.agents.integrations.vectorstores.s3vectors.S3VectorsVectorStore"
347+
334348
# MCP resource names
335349
MCP_SERVER = "flink_agents.integrations.mcp.mcp.MCPServer"

0 commit comments

Comments
 (0)