giulia-fontanella commited on
Commit
4e98072
·
verified ·
1 Parent(s): 61ff405

Update tools.py

Browse files
Files changed (1) hide show
  1. tools.py +3 -1
tools.py CHANGED
@@ -1,4 +1,5 @@
1
  import base64
 
2
  from langchain_core.messages import AnyMessage, HumanMessage, AIMessage
3
  from langchain.tools import tool
4
  from langchain_community.tools.tavily_search import TavilySearchResults
@@ -45,6 +46,7 @@ def read_python(file_path: str) -> str:
45
  return f.read()
46
  except Exception as e:
47
  return f"Error reading Python file: {str(e)}"
 
48
 
49
  class ExtractTextFromImage:
50
  def __init__(self, vision_llm):
@@ -178,7 +180,7 @@ def web_search(query: str) -> str:
178
 
179
  Args:
180
  query: The search query."""
181
- search_docs = TavilySearchResults(max_results=3).invoke(query=query)
182
  formatted_search_docs = "\n\n---\n\n".join(
183
  [
184
  f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'
 
1
  import base64
2
+ import pandas as pd
3
  from langchain_core.messages import AnyMessage, HumanMessage, AIMessage
4
  from langchain.tools import tool
5
  from langchain_community.tools.tavily_search import TavilySearchResults
 
46
  return f.read()
47
  except Exception as e:
48
  return f"Error reading Python file: {str(e)}"
49
+
50
 
51
  class ExtractTextFromImage:
52
  def __init__(self, vision_llm):
 
180
 
181
  Args:
182
  query: The search query."""
183
+ search_docs = TavilySearchResults(max_results=3).invoke(query)
184
  formatted_search_docs = "\n\n---\n\n".join(
185
  [
186
  f'<Document source="{doc.metadata["source"]}" page="{doc.metadata.get("page", "")}"/>\n{doc.page_content}\n</Document>'