refactor llm/embedding flow
Browse files- app/gemini_client.py +1 -1
app/gemini_client.py
CHANGED
|
@@ -36,7 +36,7 @@ class GeminiClient:
|
|
| 36 |
response = embed_content(
|
| 37 |
model=self.model,
|
| 38 |
content=text,
|
| 39 |
-
task_type="
|
| 40 |
)
|
| 41 |
logger.info(f"[GEMINI][EMBEDDING][RAW_RESPONSE] {response}")
|
| 42 |
return response['embedding']
|
|
|
|
| 36 |
response = embed_content(
|
| 37 |
model=self.model,
|
| 38 |
content=text,
|
| 39 |
+
task_type="retrieval_query"
|
| 40 |
)
|
| 41 |
logger.info(f"[GEMINI][EMBEDDING][RAW_RESPONSE] {response}")
|
| 42 |
return response['embedding']
|