Merge pull request #231 from restackio/make-onboarding-better

leawn · web-flow · commit fd8c76e772ed · 2025-05-12T15:53:10.000+02:00
Make onboarding better
diff --git a/agent_apis/.env.Example b/agent_apis/.env.Example
@@ -1,8 +1 @@
-OPENAI_API_KEY=<your-openai-api-key>
-
-# Restack Cloud (Optional)
-
-# RESTACK_ENGINE_ID=<your-engine-id>
-# RESTACK_ENGINE_API_KEY=<your-engine-api-key>
-# RESTACK_ENGINE_ADDRESS=<your-engine-address>
-# RESTACK_CLOUD_TOKEN=<your-cloud-token>
+OPENAI_API_KEY=<your-openai-api-key>
diff --git a/agent_apis/src/functions/llm.py b/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:
diff --git a/agent_chat/.env.Example b/agent_chat/.env.Example
@@ -1,8 +1 @@
-# Restack Cloud (Optional)
-
-# RESTACK_ENGINE_ID=<your-engine-id>
-# RESTACK_ENGINE_API_KEY=<your-engine-api-key>
-# RESTACK_ENGINE_API_ADDRESS=<your-engine-api-address>
-# RESTACK_ENGINE_ADDRESS=<your-engine-address>
-
-# RESTACK_API_KEY=<your-api-key>
+OPENAI_API_KEY=<your-openai-api-key>
diff --git a/agent_chat/src/functions/llm_chat.py b/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(
diff --git a/agent_humanloop/.env.Example b/agent_humanloop/.env.Example
@@ -1,6 +1 @@
-# Restack Cloud (Optional)
-
-# RESTACK_ENGINE_ID=<your-engine-id>
-# RESTACK_ENGINE_API_KEY=<your-engine-api-key>
-# RESTACK_ENGINE_ADDRESS=<your-engine-address>
-# RESTACK_CLOUD_TOKEN=<your-cloud-token>
+OPENAI_API_KEY=<your-openai-api-key>
diff --git a/agent_rag/src/functions/llm_chat.py b/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(
diff --git a/agent_stream/.env.Example b/agent_stream/.env.Example
@@ -1,8 +1 @@
-# Restack Cloud (Optional)
-
-# RESTACK_ENGINE_ID=<your-engine-id>
-# RESTACK_ENGINE_API_KEY=<your-engine-api-key>
-# RESTACK_ENGINE_API_ADDRESS=<your-engine-api-address>
-# RESTACK_ENGINE_ADDRESS=<your-engine-address>
-
-# RESTACK_API_KEY=<your-api-key>
+OPENAI_API_KEY=<your-openai-api-key>
diff --git a/agent_stream/src/functions/llm_chat.py b/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
diff --git a/agent_telephony/twilio_livekit/agent_twilio/.env.Example b/agent_telephony/twilio_livekit/agent_twilio/.env.Example
@@ -1,5 +1,3 @@
-
-RESTACK_API_KEY=
 OPENAI_API_KEY=
 
 LIVEKIT_API_KEY=
@@ -15,13 +13,4 @@ ELEVEN_API_KEY=
 DEEPGRAM_API_KEY=
 OPENAI_API_KEY=
 
-GCP_CREDENTIALS=
-
-# Restack Cloud (Optional)
-
-# RESTACK_ENGINE_ID=<your-engine-id>
-# RESTACK_ENGINE_API_KEY=<your-engine-api-key>
-# RESTACK_ENGINE_API_ADDRESS=<your-engine-api-address>
-# RESTACK_ENGINE_ADDRESS=<your-engine-address>
-
-
+GCP_CREDENTIALS=
diff --git a/agent_telephony/vapi/agent_vapi/.env.Example b/agent_telephony/vapi/agent_vapi/.env.Example
@@ -1,12 +1,2 @@
-
-RESTACK_API_KEY=
-VAPI_TOKEN=
-
-# Restack Cloud (Optional)
-
-# RESTACK_ENGINE_ID=<your-engine-id>
-# RESTACK_ENGINE_API_KEY=<your-engine-api-key>
-# RESTACK_ENGINE_API_ADDRESS=<your-engine-api-address>
-# RESTACK_ENGINE_ADDRESS=<your-engine-address>
-
-
+OPENAI_API_KEY=
+VAPI_TOKEN=
diff --git a/agent_telephony/vapi/agent_vapi/src/functions/llm_chat.py b/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
diff --git a/agent_todo/.env.Example b/agent_todo/.env.Example
@@ -1,7 +1 @@
-# Restack Cloud (Optional)
-
-# RESTACK_ENGINE_ID=<your-engine-id>
-# RESTACK_ENGINE_API_KEY=<your-engine-api-key>
-# RESTACK_ENGINE_API_ADDRESS=<your-engine-api-address>
-# RESTACK_ENGINE_ADDRESS=<your-engine-address>
-# RESTACK_CLOUD_TOKEN=<your-cloud-token>
+OPENAI_API_KEY=<your-openai-api-key>
diff --git a/agent_todo/src/functions/llm_chat.py b/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)
 
diff --git a/agent_tool/src/functions/llm_chat.py b/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)
 
diff --git a/agent_video/src/functions/llm_chat.py b/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
diff --git a/agent_voice/livekit/agent/.env.example b/agent_voice/livekit/agent/.env.example
@@ -1,4 +1,3 @@
-RESTACK_API_KEY=
 LIVEKIT_API_KEY=
 LIVEKIT_API_SECRET=
 LIVEKIT_URL="wss://XXXXlivekit.cloud"
@@ -10,11 +9,4 @@ TWILIO_TRUNK_AUTH_PASSWORD=
 
 ELEVEN_API_KEY=
 DEEPGRAM_API_KEY=
-OPENAI_API_KEY=
-
-# Restack Cloud (Optional)
-
-# RESTACK_ENGINE_ID=<your-engine-id>
-# RESTACK_ENGINE_API_KEY=<your-engine-api-key>
-# RESTACK_ENGINE_API_ADDRESS=<your-engine-api-address>
-# RESTACK_ENGINE_ADDRESS=<your-engine-address>
+OPENAI_API_KEY=
diff --git a/agent_voice/livekit/agent/src/functions/llm_chat.py b/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
diff --git a/agent_voice/livekit/livekit_pipeline/.env.example b/agent_voice/livekit/livekit_pipeline/.env.example
@@ -4,11 +4,4 @@ LIVEKIT_API_SECRET=
 
 OPENAI_API_KEY=
 DEEPGRAM_API_KEY=
-CARTESIA_API_KEY=
-
-# Restack Cloud (Optional)
-
-# RESTACK_ENGINE_ID=<your-engine-id>
-# RESTACK_ENGINE_API_KEY=<your-engine-api-key>
-# RESTACK_ENGINE_API_ADDRESS=<your-engine-api-address>
-# RESTACK_ENGINE_ADDRESS=<your-engine-address>
+CARTESIA_API_KEY=
diff --git a/encryption/.env.example b/encryption/.env.example
@@ -1,6 +1 @@
-# Restack Cloud (Optional)
-
-# RESTACK_ENGINE_ID=<your-engine-id>
-# RESTACK_ENGINE_API_KEY=<your-engine-api-key>
-# RESTACK_ENGINE_ADDRESS=<your-engine-address>
-# RESTACK_CLOUD_TOKEN=<your-cloud-token>
+OPENAI_API_KEY=<your-openai-api-key>
diff --git a/pdf_ocr/.env.Example b/pdf_ocr/.env.Example
@@ -1,9 +1 @@
-
-OPENAI_API_KEY=<your-openai-api-key>
-
-# Restack Cloud (Optional)
-
-# RESTACK_ENGINE_ID=<your-engine-id>
-# RESTACK_ENGINE_API_KEY=<your-engine-api-key>
-# RESTACK_ENGINE_ADDRESS=<your-engine-address>
-# RESTACK_CLOUD_TOKEN=<your-cloud-token>
+OPENAI_API_KEY=<your-openai-api-key>
diff --git a/pdf_ocr/src/functions/openai_chat.py b/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:
diff --git a/production_demo/.env.Example b/production_demo/.env.Example
@@ -1,6 +1 @@
-# Restack Cloud (Optional)
-
-# RESTACK_ENGINE_ID=<your-engine-id>
-# RESTACK_ENGINE_API_KEY=<your-engine-api-key>
-# RESTACK_ENGINE_ADDRESS=<your-engine-address>
-# RESTACK_ENGINE_API_ADDRESS=<your-engine-api-address>
+OPENAI_API_KEY=<your-openai-api-key>
