Fix custom tool example (#201)

* fixed readme example * added requirements
landing-ai · Aug 20, 2024 · b5faa36 · b5faa36
1 parent 955ef9e
commit b5faa36
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 2 deletions.
diff --git a/examples/custom_tools/README.md b/examples/custom_tools/README.md
@@ -22,7 +22,7 @@ call out which tool you want to use. For example:
 ```python
 import vision_agent as va
 
-agent = va.agent.VisionAgent(verbosity=2)
+agent = va.agent.VisionAgentCoder(verbosity=2)
 agent(
  "Can you use the 'template_match_' tool to find the location of pid_template.png in pid.png?",
  media="pid.png",

diff --git a/examples/custom_tools/requirements.txt b/examples/custom_tools/requirements.txt
@@ -0,0 +1,2 @@
+torch
+torchvision
diff --git a/examples/custom_tools/run_custom_tool.py b/examples/custom_tools/run_custom_tool.py
@@ -38,7 +38,7 @@ def template_match(target_image: np.ndarray, template_image: np.ndarray) -> dict
 
 
 if __name__ == "__main__":
- agent = va.agent.VisionAgent(verbosity=2)
+ agent = va.agent.VisionAgentCoder(verbosity=2)
  result = agent.chat_with_workflow(
  [
  {