Add OPENAI_API_KEY as default

Author: leawn

agent_apis/src/functions/llm.py

@@ -25,13 +25,11 @@ async def llm(function_input: FunctionInputParams) -> str:
     try:
         log.info("llm function started", input=function_input)
 
-        if os.environ.get("RESTACK_API_KEY") is None:
-            error_message = "RESTACK_API_KEY is not set"
+        if os.environ.get("OPENAI_API_KEY") is None:
+            error_message = "OPENAI_API_KEY is not set"
             raise_exception(error_message)
 
-        client = OpenAI(
-            base_url="https://ai.restack.io", api_key=os.environ.get("RESTACK_API_KEY")
-        )
+        client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
         messages = []
         if function_input.system_content:

agent_chat/src/functions/llm_chat.py

@@ -30,13 +30,11 @@ async def llm_chat(agent_input: LlmChatInput) -> dict[str, str]:
     try:
         log.info("llm_chat function started", agent_input=agent_input)
 
-        if os.environ.get("RESTACK_API_KEY") is None:
-            error_message = "RESTACK_API_KEY is not set"
+        if os.environ.get("OPENAI_API_KEY") is None:
+            error_message = "OPENAI_API_KEY is not set"
             raise_exception(error_message)
 
-        client = OpenAI(
-            base_url="https://ai.restack.io", api_key=os.environ.get("RESTACK_API_KEY")
-        )
+        client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
         if agent_input.system_content:
             agent_input.messages.append(

agent_rag/src/functions/llm_chat.py

@@ -31,13 +31,11 @@ async def llm_chat(function_input: LlmChatInput) -> ChatCompletion:
     try:
         log.info("llm_chat function started", function_input=function_input)
 
-        if os.environ.get("RESTACK_API_KEY") is None:
-            error_message = "RESTACK_API_KEY is not set"
+        if os.environ.get("OPENAI_API_KEY") is None:
+            error_message = "OPENAI_API_KEY is not set"
             raise_exception(error_message)
 
-        client = OpenAI(
-            base_url="https://ai.restack.io", api_key=os.environ.get("RESTACK_API_KEY")
-        )
+        client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
         if function_input.system_content:
             function_input.messages.append(

agent_stream/src/functions/llm_chat.py

@@ -26,9 +26,7 @@ class LlmChatInput(BaseModel):
 @function.defn()
 async def llm_chat(function_input: LlmChatInput) -> str:
     try:
-        client = OpenAI(
-            base_url="https://ai.restack.io", api_key=os.environ.get("RESTACK_API_KEY")
-        )
+        client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
         if function_input.system_content:
             # Insert the system message at the beginning

agent_telephony/vapi/agent_vapi/src/functions/llm_chat.py

@@ -26,9 +26,7 @@ class LlmChatInput(BaseModel):
 @function.defn()
 async def llm_chat(function_input: LlmChatInput) -> str:
     try:
-        client = OpenAI(
-            base_url="https://ai.restack.io", api_key=os.environ.get("RESTACK_API_KEY")
-        )
+        client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
         if function_input.system_content:
             # Insert the system message at the beginning

agent_todo/src/functions/llm_chat.py

@@ -43,9 +43,7 @@ async def llm_chat(function_input: LlmChatInput) -> ChatCompletion:
         if os.environ.get("RESTACK_API_KEY") is None:
             raise_exception("RESTACK_API_KEY is not set")
 
-        client = OpenAI(
-            base_url="https://ai.restack.io", api_key=os.environ.get("RESTACK_API_KEY")
-        )
+        client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
         log.info("pydantic_function_tool", tools=function_input.tools)
 

agent_tool/src/functions/llm_chat.py

@@ -40,12 +40,10 @@ async def llm_chat(function_input: LlmChatInput) -> ChatCompletion:
     try:
         log.info("llm_chat function started", function_input=function_input)
 
-        if os.environ.get("RESTACK_API_KEY") is None:
-            raise_exception("RESTACK_API_KEY is not set")
+        if os.environ.get("OPENAI_API_KEY") is None:
+            raise_exception("OPENAI_API_KEY is not set")
 
-        client = OpenAI(
-            base_url="https://ai.restack.io", api_key=os.environ.get("RESTACK_API_KEY")
-        )
+        client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
         log.info("pydantic_function_tool", tools=function_input.tools)
 

agent_video/src/functions/llm_chat.py

@@ -26,9 +26,7 @@ class LlmChatInput(BaseModel):
 @function.defn()
 async def llm_chat(function_input: LlmChatInput) -> str:
     try:
-        client = OpenAI(
-            base_url="https://ai.restack.io", api_key=os.environ.get("RESTACK_API_KEY")
-        )
+        client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
         if function_input.system_content:
             # Insert the system message at the beginning

agent_voice/livekit/agent/src/functions/llm_chat.py

@@ -26,9 +26,7 @@ class LlmChatInput(BaseModel):
 @function.defn()
 async def llm_chat(function_input: LlmChatInput) -> str:
     try:
-        client = OpenAI(
-            base_url="https://ai.restack.io", api_key=os.environ.get("RESTACK_API_KEY")
-        )
+        client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
         if function_input.system_content:
             # Insert the system message at the beginning

pdf_ocr/src/functions/openai_chat.py

@@ -17,10 +17,10 @@ async def openai_chat(input: OpenAiChatInput) -> str:
     try:
         log.info("openai_chat function started", input=input)
 
-        if (os.environ.get("RESTACK_API_KEY") is None):
-            raise NonRetryableError("RESTACK_API_KEY is not set")
+        if (os.environ.get("OPENAI_API_KEY") is None):
+            raise NonRetryableError("OPENAI_API_KEY is not set")
 
-        client = OpenAI(base_url="https://ai.restack.io", api_key=os.environ.get("RESTACK_API_KEY"))
+        client = OpenAI(api_key=os.environ.get("OPENAI_API_KEY"))
 
         messages = []
         if input.system_content: