diff --git a/dsLightRag/Test/TestGoogleGemini.py b/dsLightRag/Test/TestGoogleGemini.py index eb3380d7..6034300c 100644 --- a/dsLightRag/Test/TestGoogleGemini.py +++ b/dsLightRag/Test/TestGoogleGemini.py @@ -5,8 +5,8 @@ import requests from Config.Config import GPTNB_API_KEY class ModelInteractor: - def __init__(self, api_url="https://goapi.gptnb.ai/v1/chat/completions"): - self.api_url = api_url + def __init__(self): + self.api_url = "https://goapi.gptnb.ai/v1/chat/completions", self.headers = { "Content-Type": "application/json", "Authorization": f"Bearer {GPTNB_API_KEY}"