Merge pull request #309 from PrefectHQ/gemini

Improve gemini support
PrefectHQ · Sep 13, 2024 · 4a22fb8 · 4a22fb8
2 parents ad2b455 + c3046bf
commit 4a22fb8
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/controlflow/tools/tools.py b/src/controlflow/tools/tools.py
@@ -51,10 +51,11 @@ class Tool(ControlFlowModel):
     metadata: dict = {}
 
     fn: Callable = Field(None, exclude=True)
+    _lc_tool: Optional[langchain_core.tools.BaseTool] = None
 
     def to_lc_tool(self) -> dict:
         payload = self.model_dump(include={"name", "description", "parameters"})
-        return dict(type="function", function=payload)
+        return payload
 
     @prefect_task(task_run_name="Tool call: {self.name}")
     def run(self, input: dict):