Openai backend fix (#97)

Farama-Foundation · Nov 22, 2023 · 49f1874 · 49f1874
2 parents 7975a20 + 9a05a4d
commit 49f1874
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/chatarena/backends/openai.py b/chatarena/backends/openai.py
@@ -80,7 +80,7 @@ def _get_response(self, messages):
             stop=STOP,
         )
 
-        response = completion.choices[0]["message"]["content"]
+        response = completion.choices[0].message.content
         response = response.strip()
         return response
 

diff --git a/pyproject.toml b/pyproject.toml
@@ -47,7 +47,7 @@ all_backends = ["anthropic>=0.2.8", "cohere>=4.3.1", "transformers>=4.27.4", "ba
 all_envs = ["pettingzoo>=1.24.0", "chess==1.9.4", "rlcard==1.0.5", "pygame==2.3.0", "langchain>=0.0.135"]
 database = ["supabase==2.0.3"]
 testing = ["deptry>=0.12.0", "pytest>=7.4.3", "pytest-cov>=4.1.0", "pytest-xdist>=3.4.0"]
-all = ["anthropic>=0.2.8", "cohere>=4.3.1", "transformers>=4.27.4", "gradio==3.34.0", "pydantic==1.10.13", "pettingzoo>=1.24.0", "chess==1.9.4", "rlcard==1.0.5", "pygame==2.3.0", "gymnasium>=0.28.1",
+all = ["anthropic==0.2.8", "cohere==4.3.1", "transformers>=4.27.4", "gradio==3.34.0", "pydantic==1.10.13", "pettingzoo>=1.24.0", "chess==1.9.4", "rlcard==1.0.5", "pygame==2.3.0", "gymnasium>=0.28.1",
        "supabase==2.0.3", "bardapi==0.1.11", "langchain>=0.0.135", "deptry>=0.12.0", "pytest>=7.4.3", "pytest-cov>=4.1.0", "pytest-xdist>=3.4.0"]
 
 [tool.deptry.per_rule_ignores]