Merge pull request #1546 from phidatahq/gemini-2-flash-phi-2171

gemini-2-flash-phi-2171
phidatahq · Dec 11, 2024 · 958c0fa · 958c0fa
2 parents 78cbcbf + 64cb59f
commit 958c0fa
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 3 deletions.
diff --git a/cookbook/providers/google/agent.py b/cookbook/providers/google/agent.py
@@ -5,7 +5,7 @@
 from phi.tools.yfinance import YFinanceTools
 
 agent = Agent(
-    model=Gemini(id="gemini-1.5-flash"),
+    model=Gemini(id="gemini-2.0-flash-exp"),
     tools=[YFinanceTools(stock_price=True)],
     show_tool_calls=True,
     markdown=True,

diff --git a/cookbook/providers/google/agent_stream.py b/cookbook/providers/google/agent_stream.py
@@ -6,7 +6,7 @@
 from phi.tools.yfinance import YFinanceTools
 
 agent = Agent(
-    model=Gemini(id="gemini-1.5-flash"),
+    model=Gemini(id="gemini-2.0-flash-exp"),
     tools=[YFinanceTools(stock_price=True)],
     instructions=["Use tables where possible."],
     markdown=True,

diff --git a/phi/model/google/gemini.py b/phi/model/google/gemini.py
@@ -420,13 +420,25 @@ def _format_function_call_results(
             messages (List[Message]): The list of conversation messages.
         """
         if function_call_results:
+            combined_content = []  # Use a list to collect all result contents
+            combined_parts = []  # Use a list to collect all function responses
+
             for result in function_call_results:
                 s = Struct()
                 s.update({"result": [result.content]})
                 function_response = genai.protos.Part(
                     function_response=genai.protos.FunctionResponse(name=result.tool_name, response=s)
                 )
-                messages.append(Message(role="tool", content=result.content, parts=[function_response]))
+                combined_content.append(result.content)
+                combined_parts.append(function_response)
+
+            messages.append(
+                Message(
+                    role="tool",
+                    content="\n".join(combined_content),
+                    parts=combined_parts
+                )
+            )
 
     def _handle_tool_calls(self, assistant_message: Message, messages: List[Message], model_response: ModelResponse):
         """