Update tools.py
Browse files
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
|
| 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>'
|