From 780e3aa304bde5abb07e044c3c4ffb840ff8edd0 Mon Sep 17 00:00:00 2001 From: "jinli.yl" Date: Mon, 16 Jun 2025 10:21:21 +0800 Subject: [PATCH] update cookbook & es vs --- cookbook/step_agent/readme.md | 40 +++++++++++----------- experiencemaker/storage/es_vector_store.py | 8 +++++ 2 files changed, 28 insertions(+), 20 deletions(-) diff --git a/cookbook/step_agent/readme.md b/cookbook/step_agent/readme.md index 9e1b3837..a4a4d288 100644 --- a/cookbook/step_agent/readme.md +++ b/cookbook/step_agent/readme.md @@ -62,16 +62,16 @@ python examples.py summarizer_request = { "trajectories": [successful_trajectory, failed_trajectory], "workspace_id": "my_workspace", - "metadata": { - "em_config": { - "summarizer": { - "backend": "step", - "enable_step_segmentation": True, - "enable_similarity_search": True, - "enable_experience_validation": True - } - } - } + # "metadata": { + # "em_config": { + # "summarizer": { + # "backend": "step", + # "enable_step_segmentation": True, + # "enable_similarity_search": True, + # "enable_experience_validation": True + # } + # } + # } } response = requests.post(f"{SERVICE_URL}/summarizer", json=summarizer_request) @@ -83,16 +83,16 @@ response = requests.post(f"{SERVICE_URL}/summarizer", json=summarizer_request) context_request = { "trajectory": current_trajectory, "workspace_id": "my_workspace", - "metadata": { - "em_config": { - "context_generator": { - "backend": "step", - "enable_llm_rerank": True, - "enable_context_rewrite": True, - "final_top_k": 5 - } - } - } + # "metadata": { + # "em_config": { + # "context_generator": { + # "backend": "step", + # "enable_llm_rerank": True, + # "enable_context_rewrite": True, + # "final_top_k": 5 + # } + # } + # } } response = requests.post(f"{SERVICE_URL}/context_generator", json=context_request) diff --git a/experiencemaker/storage/es_vector_store.py b/experiencemaker/storage/es_vector_store.py index 5a6bc690..1105c007 100644 --- a/experiencemaker/storage/es_vector_store.py +++ b/experiencemaker/storage/es_vector_store.py @@ -175,6 +175,10 @@ class EsVectorStore(BaseVectorStore): if index_name is None: index_name = self.index_name + if not self.exist_index(index_name): + logger.warning(f"index_name={index_name} is not exists!") + return None + try: doc = self._client.get(index=index_name, id=unique_id, **kwargs) return self.doc2node(doc) @@ -187,6 +191,10 @@ class EsVectorStore(BaseVectorStore): if index_name is None: index_name = self.index_name + if not self.exist_index(index_name): + logger.warning(f"index_name={index_name} is not exists!") + return [] + query_vector = self.embedding_model.get_embeddings(query) body = {