go
Browse files- app/gemini_client.py +1 -0
- app/recommendation_pipeline.py +1 -1
app/gemini_client.py
CHANGED
|
@@ -1,4 +1,5 @@
|
|
| 1 |
from __future__ import annotations
|
|
|
|
| 2 |
|
| 3 |
import logging
|
| 4 |
import os
|
|
|
|
| 1 |
from __future__ import annotations
|
| 2 |
+
from google.genai import types
|
| 3 |
|
| 4 |
import logging
|
| 5 |
import os
|
app/recommendation_pipeline.py
CHANGED
|
@@ -149,7 +149,7 @@ def build_query_text(persona: str, chat_history: list[dict[str, str]]) -> str:
|
|
| 149 |
def chat_rank_gemini(
|
| 150 |
*,
|
| 151 |
persona: str,
|
| 152 |
-
chat_history: list[dict[str, str]],
|
| 153 |
candidates: list[dict[str, Any]],
|
| 154 |
top_n: int,
|
| 155 |
) -> list[dict[str, Any]]:
|
|
|
|
| 149 |
def chat_rank_gemini(
|
| 150 |
*,
|
| 151 |
persona: str,
|
| 152 |
+
chat_history: list[dict[str, str]] | None = None,
|
| 153 |
candidates: list[dict[str, Any]],
|
| 154 |
top_n: int,
|
| 155 |
) -> list[dict[str, Any]]:
|