Skip to content

Commit f3b1af3

Browse files
committed
fix ci
1 parent ecd53ef commit f3b1af3

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

backend/app/rag/llms/dspy.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -31,8 +31,9 @@ def get_dspy_lm_by_llama_llm(llama_llm: BaseLLM) -> dspy.LM:
3131
case "GenAI":
3232
if "models/" in llama_llm.model:
3333
# For Gemini
34+
model_name = llama_llm.model.split("models/")[1]
3435
return dspy.LM(
35-
model=f"gemini/{llama_llm.model.split("models/")[1]}",
36+
model=f"gemini/{model_name}",
3637
max_tokens=llama_llm._max_tokens,
3738
api_key=llama_llm._client._api_client.api_key,
3839
)

0 commit comments

Comments
 (0)