From 7117790f419b4484b6cb97669d5b7b843e6a82fd Mon Sep 17 00:00:00 2001
From: aotsuki <y.aotsuki@gmail.com>
Date: Sat, 13 Apr 2024 13:22:27 +0800
Subject: [PATCH] Update free_ask_internet.py

fix local llm always using CUSTOM as api token
---
 free_ask_internet.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/free_ask_internet.py b/free_ask_internet.py
index 62d9f29..37b4360 100644
--- a/free_ask_internet.py
+++ b/free_ask_internet.py
@@ -189,7 +189,7 @@ def chat(prompt, model:str,llm_auth_token:str,llm_base_url:str,using_custom_llm=
 
     if using_custom_llm:
         openai.base_url = llm_base_url
-        openai.api_key = "CUSTOM"
+        openai.api_key = llm_auth_token
 
 
     total_content = ""
@@ -239,4 +239,4 @@ def ask_internet(query:str,  debug=False):
             yield "*[{}. {}]({})*".format(str(count),url,url )  
             yield "\n"
             count += 1
- 
\ No newline at end of file
+