fix bug (#75)

* fix bug

* fix error

Author: GuanyiLi-Craig

grafi/agents/react_agent.py

@@ -45,14 +45,11 @@ class ReActAgent(Assistant):
     name: str = Field(default="ReActAgent")
     type: str = Field(default="ReActAgent")
     api_key: Optional[str] = Field(default_factory=lambda: os.getenv("OPENAI_API_KEY"))
-    tavily_api_key: Optional[str] = Field(
-        default_factory=lambda: os.getenv("TAVILY_API_KEY")
-    )
     system_prompt: Optional[str] = Field(default=AGENT_SYSTEM_MESSAGE)
     function_call_tool: FunctionCallTool = Field(
         default=TavilyTool.builder()
         .name("TavilyTestTool")
-        .api_key(tavily_api_key)
+        .api_key(os.getenv("TAVILY_API_KEY"))
         .max_tokens(6000)
         .search_depth("advanced")
         .build()

tests/agents/test_react_agent.py

@@ -6,16 +6,15 @@
 
 import pytest
 
+# Set a dummy TAVILY_API_KEY to prevent initialization errors
+os.environ["TAVILY_API_KEY"] = "test-tavily-key"
+
 # Now import after setting the environment variable
 from grafi.agents.react_agent import ReActAgent
 from grafi.agents.react_agent import ReActAgentBuilder
 from grafi.tools.function_calls.impl.tavily_tool import TavilyTool
 
 
-# Set environment variable before any imports to prevent API key errors
-# os.environ["TAVILY_API_KEY"] = "mock-api-key-for-testing"
-
-
 @pytest.fixture(autouse=True)
 def mock_tavily_client():
     """Automatically mock TavilyClient for all tests."""