.github fix (#2)

* .github fix * return-type type-hint
opea-project · Dec 9, 2024 · 36bdd51 · 36bdd51
1 parent 2120c93
commit 36bdd51
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 2 deletions.
diff --git a/.github/workflows/docker/compose/dataprep-compose.yaml b/.github/workflows/docker/compose/dataprep-compose.yaml
@@ -63,3 +63,7 @@ services:
     build:
       dockerfile: comps/dataprep/multimedia2text/audio2text/Dockerfile
     image: ${REGISTRY:-opea}/dataprep-audio2text:${TAG:-latest}
+  dataprep-elasticsearch:
+    build:
+      dockerfile: comps/dataprep/elasticsearch/langchain/Dockerfile
+    image: ${REGISTRY:-opea}/dataprep-elasticsearch:${TAG:-latest}
diff --git a/.github/workflows/docker/compose/retrievers-compose.yaml b/.github/workflows/docker/compose/retrievers-compose.yaml
@@ -47,3 +47,7 @@ services:
     build:
       dockerfile: comps/retrievers/neo4j/llama_index/Dockerfile
     image: ${REGISTRY:-opea}/retriever-neo4j-llamaindex:${TAG:-latest}
+  retriever-elasticsearch:
+    build:
+      dockerfile: comps/retrievers/elasticsearch/langchain/Dockerfile
+    image: ${REGISTRY:-opea}/retriever-elasticsearch:${TAG:-latest}
diff --git a/comps/dataprep/elasticsearch/langchain/README.md b/comps/dataprep/elasticsearch/langchain/README.md
@@ -17,7 +17,7 @@ export INDEX_NAME=${your_index_name}
 
 ### 1.3 Start Elasticsearch
 
-Please refer to this [readme](../../../vectorstores/Elasticsearch/README.md).
+Please refer to this [readme](../../../vectorstores/elasticsearch/README.md).
 
 ### 1.4 Start Document Preparation Microservice for Elasticsearch with Python Script
 

diff --git a/comps/retrievers/elasticsearch/langchain/retriever_elasticsearch.py b/comps/retrievers/elasticsearch/langchain/retriever_elasticsearch.py
@@ -53,7 +53,7 @@ def get_elastic_store(embedder: Union[HuggingFaceEndpointEmbeddings, HuggingFace
     port=7000,
 )
 @register_statistics(names=["opea_service@retriever_elasticsearch"])
-async def retrieve(input: EmbedDoc) -> list:
+async def retrieve(input: EmbedDoc) -> SearchedDoc:
     """Retrieve documents based on similarity search type."""
     if LOG_FLAG:
         logger.info(input)