nexusbert commited on
Commit
4989e91
·
1 Parent(s): e9fab11
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]]: