diff --git a/application/llm/google_ai.py b/application/llm/google_ai.py index 31943601..48254349 100644 --- a/application/llm/google_ai.py +++ b/application/llm/google_ai.py @@ -22,7 +22,7 @@ class GoogleLLM(BaseLLM): parts = [] if role and content is not None: if isinstance(content, str): - parts = [types.Part.from_text(content)] + parts = [types.Part.from_text(text=content)] elif isinstance(content, list): for item in content: if "text" in item: