update cookbook & es vs

This commit is contained in:
jinli.yl 2025-06-16 10:21:21 +08:00
parent 4b06330270
commit 780e3aa304
2 changed files with 28 additions and 20 deletions

View file

@ -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)

View file

@ -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 = {