agent: fix tools setup

ggerganov · Oct 25, 2024 · 30bd00b · 30bd00b
1 parent 5c414a3
commit 30bd00b
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 3 deletions.
diff --git a/examples/agent/Dockerfile.tools b/examples/agent/Dockerfile.tools
@@ -3,7 +3,8 @@ FROM python:3.12-slim
 RUN python -m pip install --upgrade pip && \
     apt clean cache
 
-COPY requirements.txt tools/*.py /root/
+COPY requirements.txt /root/
+COPY tools /root/tools
 WORKDIR /root
 RUN pip install -r requirements.txt
 

diff --git a/examples/agent/tools/search.py b/examples/agent/tools/search.py
@@ -70,6 +70,7 @@ def extract_results(search_response):
     if not response.ok:
         raise Exception(response.text)
     response.raise_for_status()
-    results = list(itertools.islice(extract_results(response.json()), max_results))
-    print(json.dumps(dict(query=query, response=response, results=results), indent=2))
+    response_json = response.json()
+    results = list(itertools.islice(extract_results(response_json), max_results))
+    print(json.dumps(dict(query=query, response=response_json, results=results), indent=2))
     return results