diff --git a/src/pipeline/core/es_client.py b/src/pipeline/core/es_client.py index 1f01dde..1718597 100644 --- a/src/pipeline/core/es_client.py +++ b/src/pipeline/core/es_client.py @@ -1,11 +1,10 @@ import httpx from pipeline.config import config - class AsyncES: def __init__(self): self.base = f"{config['es_host']}:{config['es_port']}" - self.auth = (config["es_user"], config["es_password"]) + self.auth = (config["es_user"], config['es_password']) async def create_index(self, index: str): """ @@ -16,7 +15,7 @@ class AsyncES: f"{self.base}/{index}", auth=self.auth, json={ - "settings": {"analysis": {"analyzer": {"default": {"type": "ik_max_word"}}}}, + "settings": {"analysis": {"analyzer": {"default": {"type": "standard"}}}}, "mappings": { "properties": { "text": {"type": "text"}, @@ -43,5 +42,4 @@ class AsyncES: resp = await client.post(f"{self.base}/{index}/_search", auth=self.auth, json={"size": top_k, "query": {"match": {"text": query}}}) return resp.json() - es_client = AsyncES