fix(agent) migrate from azure to base openai

plastic-labs · Oct 16, 2024 · 6a8a189 · 6a8a189
1 parent 4f8be70
commit 6a8a189
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 21 deletions.
diff --git a/agent/agent/chain.py b/agent/agent/chain.py
@@ -1,7 +1,7 @@
 from os import getenv
 from typing import List
 
-from openai import AzureOpenAI, completions
+from openai import OpenAI
 from dotenv import load_dotenv
 
 from honcho import Honcho
@@ -30,13 +30,9 @@ def __init__(
 
     model_config = ConfigDict(arbitrary_types_allowed=True)
 
-    openai = AzureOpenAI(
-        api_key=getenv("AZURE_OPENAI_API_KEY", "placeholder"),
-        azure_endpoint=getenv("AZURE_OPENAI_ENDPOINT", "placeholder"),
-        api_version=getenv("AZURE_OPENAI_API_VERSION", "2024-02-01"),
-    )
+    openai = OpenAI()
 
-    model = getenv("AZURE_OPENAI_DEPLOYMENT", "placeholder")
+    model = "gpt-4o"
 
 
 class ThinkCall(HonchoCall):
@@ -84,14 +80,14 @@ def history(self) -> str:
 
     def call(self):
         response = self.openai.chat.completions.create(
-            model=getenv("AZURE_OPENAI_DEPLOYMENT", "placeholder"),
+            model=self.model,
             messages=[self.template(), {"role": "user", "content": self.user_input}],
         )
         return response.choices[0].message
 
     def stream(self):
         completion = self.openai.chat.completions.create(
-            model=getenv("AZURE_OPENAI_DEPLOYMENT", "placeholder"),
+            model=self.model,
             messages=[self.template(), {"role": "user", "content": self.user_input}],
             stream=True,
         )
@@ -136,14 +132,14 @@ def history(self) -> List[dict]:
 
     def call(self):
         response = self.openai.chat.completions.create(
-            model=getenv("AZURE_OPENAI_DEPLOYMENT", "placeholder"),
+            model=self.model,
             messages=self.template(),
         )
         return response.choices[0].message
 
     def stream(self):
         completion = self.openai.chat.completions.create(
-            model=getenv("AZURE_OPENAI_DEPLOYMENT", "placeholder"),
+            model=self.model,
             messages=self.template(),
             stream=True,
         )

diff --git a/api/.env.template b/api/.env.template
@@ -1,8 +1,5 @@
-# Azure Mirascope Keys
-AZURE_OPENAI_ENDPOINT=
-AZURE_OPENAI_API_KEY=
-AZURE_OPENAI_API_VERSION=
-AZURE_OPENAI_DEPLOYMENT=
+# OpenAI Keys
+OPENAI_API_KEY=
 
 # NextJS & fastAPI
 URL=http://localhost:3000

diff --git a/bot/.env.template b/bot/.env.template
@@ -1,8 +1,5 @@
-# Azure Mirascope Keys
-AZURE_OPENAI_ENDPOINT=
-AZURE_OPENAI_API_KEY=
-AZURE_OPENAI_API_VERSION=
-AZURE_OPENAI_DEPLOYMENT=
+# OpenAI Keys
+OPENAI_API_KEY=
 
 # Discord Settings
 BOT_TOKEN=