🔒 remove url to env

bruceunx · May 3, 2024 · 01d4465 · 01d4465
1 parent 63ad40b
commit 01d4465
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 5 deletions.
diff --git a/gpt_server/routers/cohere.py b/gpt_server/routers/cohere.py
@@ -12,7 +12,7 @@
 
 api_key = os.environ['COHERE_KEY']
 
-url = "https://api.cohere.ai/v1/chat"
+url = os.environ['COHERE_URL']
 
 headers = {
     "Authorization": f"Bearer {api_key}",

diff --git a/gpt_server/routers/gemini.py b/gpt_server/routers/gemini.py
@@ -16,8 +16,8 @@
 
 api_key = os.environ['GEMINI_KEY']
 
-API_URL = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-pro:streamGenerateContent?alt=sse&key={api_key}"
-API_URL_PRO = f"https://generativelanguage.googleapis.com/v1beta/models/gemini-1.5-pro-latest:streamGenerateContent?alt=sse&key={api_key}"
+API_URL = f"{os.environ['GEMINI_URL']}?alt=sse&key={api_key}"
+API_URL_PRO = f"{os.environ['GEMINI_PRO_URL']}?alt=sse&key={api_key}"
 
 PROXY = os.environ['proxy'] if 'proxy' in os.environ else None
 

diff --git a/gpt_server/routers/groq.py b/gpt_server/routers/groq.py
@@ -8,7 +8,7 @@
 
 router = APIRouter()
 
-url = "https://api.groq.com/openai/v1/chat/completions"
+url = os.environ['GROQ_URL']
 
 headers = {
     "Authorization": f"Bearer {os.environ['GROQ_KEY']}",

diff --git a/gpt_server/routers/openai.py b/gpt_server/routers/openai.py
@@ -8,7 +8,7 @@
 
 router = APIRouter()
 
-url = "https://api.pumpkinaigc.online/v1/chat/completions"
+url = os.environ['OPENAI_URL']
 
 headers = {
     "Authorization": f"Bearer {os.environ['OPENAI_KEY']}",