Spaces:
Running
Running
Commit
·
b746682
1
Parent(s):
60ee681
disabling tool invocation stops sending it the tools
Browse files- src/manager/manager.py +4 -1
src/manager/manager.py
CHANGED
@@ -91,7 +91,10 @@ class GeminiManager:
|
|
91 |
max_tries=3,
|
92 |
jitter=None)
|
93 |
def generate_response(self, messages):
|
94 |
-
|
|
|
|
|
|
|
95 |
return self.client.models.generate_content_stream(
|
96 |
model=self.model_name,
|
97 |
contents=messages,
|
|
|
91 |
max_tries=3,
|
92 |
jitter=None)
|
93 |
def generate_response(self, messages):
|
94 |
+
if self.check_mode(Mode.ENABLE_TOOL_INVOCATION):
|
95 |
+
tools = self.toolsLoader.getTools()
|
96 |
+
else:
|
97 |
+
tools = []
|
98 |
return self.client.models.generate_content_stream(
|
99 |
model=self.model_name,
|
100 |
contents=messages,
|