fix openai key

gnosis · Feb 11, 2024 · 6a25d30 · 6a25d30
1 parent 882feb7
commit 6a25d30
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/prediction_market_agent/agents/autogen_agent.py b/prediction_market_agent/agents/autogen_agent.py
@@ -18,7 +18,7 @@ def get_base_llm_config(self) -> dict[str, t.Any]:
             "config_list": [
                 {
                     "model": "gpt-4",
-                    "api_key": keys.openai,
+                    "api_key": keys.openai_api_key,
                 }
             ],
             "temperature": 0,

diff --git a/prediction_market_agent/agents/langchain_agent.py b/prediction_market_agent/agents/langchain_agent.py
@@ -11,7 +11,7 @@
 class LangChainAgent(AbstractAgent):
     def __init__(self) -> None:
         keys = utils.APIKeys()
-        llm = OpenAI(openai_api_key=keys.openai)
+        llm = OpenAI(openai_api_key=keys.openai_api_key)
         # Can use pre-defined search tool
         # TODO: Tavily tool could give better results
         # https://docs.tavily.com/docs/tavily-api/langchain