Added functions for arithmetic operations in tools

pyproject.toml

@@ -0,0 +1,20 @@
+[project]
+name = "poetry-gaia"
+version = "0.1.0"
+description = "GAIA Benchmarking"
+authors = [
+    {name = "sharath", email = ""}
+]
+readme = "README.md"
+requires-python = ">=3.9"
+dependencies = [
+"gradio",
+"requests",
+"pandas",
+"numpy",
+"langgraph"
+]
+
+[build-system]
+requires = ["poetry-core>=2.0.0,<3.0.0"]
+build-backend = "poetry.core.masonry.api"

tools.py

@@ -0,0 +1,25 @@
+from typing import List, TypedDict, Annotated, Optional
+from langchain_openai import ChatOpenAI
+from langchain_core.messages import SystemMessage, HumanMessage, AnyMessage
+from langgraph.graph.message import add_messages
+from langgraph.prebuilt import ToolNode, tools_condition
+
+class AgentState(TypedDict):
+    """Agent state to be passed to the tool."""
+    messages: Annotated[List[AnyMessage], add_messages]
+    
+def add(a: int, b: int) -> int:
+    """Add two numbers."""
+    return a + b
+def subtract(a: int, b: int) -> int:
+    """Subtract two numbers."""
+    return a - b
+def multiply(a: int, b: int) -> int:
+    """Multiply two numbers."""
+    return a * b
+def divide(a: int, b: int) -> Optional[float]:
+    """Divide two numbers."""
+    if b == 0:
+        return None
+    return a / b
+