This is an automated email from the ASF dual-hosted git repository.
xtsong pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/flink-agents.git
from da299d9 [hotfix] toolCallContext not updated in ChatModelAction,
causing data loss (#289)
new 51bfde7 [hotfix] Fix CI failure due to latest version of the
onnxruntime dependency not supporting macos
new a699982 [hotfix] Improve e2e test robustness
new 90cd7c6 [Feature] Support ollama embedding models in java.
The 3 revisions listed above as "new" are entirely new to this
repository and will be described in separate emails. The revisions
listed as "add" were already present in the repository and have only
been added to this reference.
Summary of changes:
...nnection.java => EmbeddingModelConnection.java} | 11 +-
...delConnection.java => EmbeddingModelSetup.java} | 11 +-
.../model/BaseEmbeddingModelConnection.java | 83 ++++++++
.../embedding/model/BaseEmbeddingModelSetup.java | 113 +++++++++++
e2e-test/integration-test/pom.xml | 5 +
.../integration/test/AgentWithOllamaEmbedding.java | 217 +++++++++++++++++++++
...e.java => AgentWithOllamaEmbeddingExample.java} | 28 +--
.../test-scripts/test_agent_plan_compatibility.sh | 7 +-
.../test-scripts/test_java_config_in_python.sh | 7 +-
integrations/chat-models/ollama/pom.xml | 2 +-
.../ollama/pom.xml | 15 +-
.../ollama/OllamaEmbeddingModelConnection.java | 126 ++++++++++++
.../ollama/OllamaEmbeddingModelSetup.java | 43 ++--
.../ollama/OllamaEmbeddingModelConnectionTest.java | 89 +++++++++
.../{chat-models => embedding-models}/pom.xml | 6 +-
integrations/pom.xml | 5 +
.../org/apache/flink/agents/plan/AgentPlan.java | 6 +
python/pyproject.toml | 17 +-
tools/e2e.sh | 100 +++++++++-
19 files changed, 819 insertions(+), 72 deletions(-)
copy
api/src/main/java/org/apache/flink/agents/api/annotation/{ChatModelConnection.java
=> EmbeddingModelConnection.java} (73%)
copy
api/src/main/java/org/apache/flink/agents/api/annotation/{ChatModelConnection.java
=> EmbeddingModelSetup.java} (74%)
create mode 100644
api/src/main/java/org/apache/flink/agents/api/embedding/model/BaseEmbeddingModelConnection.java
create mode 100644
api/src/main/java/org/apache/flink/agents/api/embedding/model/BaseEmbeddingModelSetup.java
create mode 100644
e2e-test/integration-test/src/main/java/org/apache/flink/agents/integration/test/AgentWithOllamaEmbedding.java
copy
e2e-test/integration-test/src/main/java/org/apache/flink/agents/integration/test/{AgentWithOllamaExample.java
=> AgentWithOllamaEmbeddingExample.java} (66%)
copy integrations/{chat-models => embedding-models}/ollama/pom.xml (78%)
create mode 100644
integrations/embedding-models/ollama/src/main/java/org/apache/flink/agents/integrations/embeddingmodels/ollama/OllamaEmbeddingModelConnection.java
copy
api/src/main/java/org/apache/flink/agents/api/chat/model/BaseChatModelConnection.java
=>
integrations/embedding-models/ollama/src/main/java/org/apache/flink/agents/integrations/embeddingmodels/ollama/OllamaEmbeddingModelSetup.java
(53%)
create mode 100644
integrations/embedding-models/ollama/src/test/java/org/apache/flink/agents/integrations/embeddingmodels/ollama/OllamaEmbeddingModelConnectionTest.java
copy integrations/{chat-models => embedding-models}/pom.xml (90%)
