huanghai/dsProject
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

'commit'

Browse Source
This commit is contained in:
HuangHai
2025-08-15 16:17:26 +08:00
parent b5bb154dcf
commit b136062c28
1 changed files with 16 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
dsLightRag/Test/TestGoogleGemini.py
View File

@@ -4,12 +4,14 @@ import requests
from Config.Config import GPTNB_API_KEY from Config.Config import GPTNB_API_KEY
class ModelInteractor: class ModelInteractor:
def __init__(self): def __init__(self, api_key=GPTNB_API_KEY, api_url="https://goapi.gptnb.ai/v1/chat/completions"):
self.api_url = "https://goapi.gptnb.ai/v1/chat/completions", self.api_key = api_key
self.api_url = api_url
self.headers = { self.headers = {
"Content-Type": "application/json", "Content-Type": "application/json",
"Authorization": f"Bearer {GPTNB_API_KEY}" "Authorization": f"Bearer {self.api_key}"
} }
def stream_request(self, model, prompt, temperature=0.7): def stream_request(self, model, prompt, temperature=0.7):
@@ -74,6 +76,7 @@ class ModelInteractor:
except requests.exceptions.RequestException as e: except requests.exceptions.RequestException as e:
print(f"请求发生错误: {e}") print(f"请求发生错误: {e}")
# 示例使用 # 示例使用
if __name__ == "__main__": if __name__ == "__main__":
# 创建模型交互器实例 # 创建模型交互器实例