diff --git a/api/src/main/java/org/apache/flink/agents/api/resource/ResourceName.java b/api/src/main/java/org/apache/flink/agents/api/resource/ResourceName.java index e4002be7c..0d6ee413e 100644 --- a/api/src/main/java/org/apache/flink/agents/api/resource/ResourceName.java +++ b/api/src/main/java/org/apache/flink/agents/api/resource/ResourceName.java @@ -59,6 +59,12 @@ public static final class ChatModel { public static final String AZURE_SETUP = "org.apache.flink.agents.integrations.chatmodels.azureai.AzureAIChatModelSetup"; + // Bedrock + public static final String BEDROCK_CONNECTION = + "org.apache.flink.agents.integrations.chatmodels.bedrock.BedrockChatModelConnection"; + public static final String BEDROCK_SETUP = + "org.apache.flink.agents.integrations.chatmodels.bedrock.BedrockChatModelSetup"; + // Ollama public static final String OLLAMA_CONNECTION = "org.apache.flink.agents.integrations.chatmodels.ollama.OllamaChatModelConnection"; @@ -131,6 +137,12 @@ public static final class EmbeddingModel { public static final String OLLAMA_SETUP = "org.apache.flink.agents.integrations.embeddingmodels.ollama.OllamaEmbeddingModelSetup"; + // Bedrock + public static final String BEDROCK_CONNECTION = + "org.apache.flink.agents.integrations.embeddingmodels.bedrock.BedrockEmbeddingModelConnection"; + public static final String BEDROCK_SETUP = + "org.apache.flink.agents.integrations.embeddingmodels.bedrock.BedrockEmbeddingModelSetup"; + // Python Wrapper public static final String PYTHON_WRAPPER_CONNECTION = "org.apache.flink.agents.api.embedding.model.python.PythonEmbeddingModelConnection"; @@ -171,6 +183,14 @@ public static final class VectorStore { public static final String ELASTICSEARCH_VECTOR_STORE = "org.apache.flink.agents.integrations.vectorstores.elasticsearch.ElasticsearchVectorStore"; + // Amazon OpenSearch (Serverless or Service domains) + public static final String OPENSEARCH_VECTOR_STORE = + "org.apache.flink.agents.integrations.vectorstores.opensearch.OpenSearchVectorStore"; + + // Amazon S3 Vectors + public static final String S3_VECTORS_VECTOR_STORE = + "org.apache.flink.agents.integrations.vectorstores.s3vectors.S3VectorsVectorStore"; + // Python Wrapper public static final String PYTHON_WRAPPER_VECTOR_STORE = "org.apache.flink.agents.api.vectorstores.python.PythonVectorStore"; diff --git a/python/flink_agents/api/resource.py b/python/flink_agents/api/resource.py index f8d7741a0..f481edcf1 100644 --- a/python/flink_agents/api/resource.py +++ b/python/flink_agents/api/resource.py @@ -276,6 +276,10 @@ class Java: AZURE_CONNECTION = "org.apache.flink.agents.integrations.chatmodels.azureai.AzureAIChatModelConnection" AZURE_SETUP = "org.apache.flink.agents.integrations.chatmodels.azureai.AzureAIChatModelSetup" + # Bedrock + BEDROCK_CONNECTION = "org.apache.flink.agents.integrations.chatmodels.bedrock.BedrockChatModelConnection" + BEDROCK_SETUP = "org.apache.flink.agents.integrations.chatmodels.bedrock.BedrockChatModelSetup" + # Ollama OLLAMA_CONNECTION = "org.apache.flink.agents.integrations.chatmodels.ollama.OllamaChatModelConnection" OLLAMA_SETUP = "org.apache.flink.agents.integrations.chatmodels.ollama.OllamaChatModelSetup" @@ -313,6 +317,10 @@ class Java: OLLAMA_CONNECTION = "org.apache.flink.agents.integrations.embeddingmodels.ollama.OllamaEmbeddingModelConnection" OLLAMA_SETUP = "org.apache.flink.agents.integrations.embeddingmodels.ollama.OllamaEmbeddingModelSetup" + # Bedrock + BEDROCK_CONNECTION = "org.apache.flink.agents.integrations.embeddingmodels.bedrock.BedrockEmbeddingModelConnection" + BEDROCK_SETUP = "org.apache.flink.agents.integrations.embeddingmodels.bedrock.BedrockEmbeddingModelSetup" + class VectorStore: """VectorStore resource names.""" @@ -331,5 +339,11 @@ class Java: # Elasticsearch ELASTICSEARCH_VECTOR_STORE = "org.apache.flink.agents.integrations.vectorstores.elasticsearch.ElasticsearchVectorStore" + # Amazon OpenSearch (Serverless or Service domains) + OPENSEARCH_VECTOR_STORE = "org.apache.flink.agents.integrations.vectorstores.opensearch.OpenSearchVectorStore" + + # Amazon S3 Vectors + S3_VECTORS_VECTOR_STORE = "org.apache.flink.agents.integrations.vectorstores.s3vectors.S3VectorsVectorStore" + # MCP resource names MCP_SERVER = "flink_agents.integrations.mcp.mcp.MCPServer"