diff --git a/dsLightRag/Test/TestGoogleGemini.py b/dsLightRag/Test/TestGoogleGemini.py index 1b03af53..6b7ca949 100644 --- a/dsLightRag/Test/TestGoogleGemini.py +++ b/dsLightRag/Test/TestGoogleGemini.py @@ -3,7 +3,7 @@ import json # API配置 API_URL = "https://goapi.gptnb.ai/v1/chat/completions" -API_KEY = "sk-amQHwiEzPIZIB2KuF5A10dC23a0e4b02B48a7a2b6aFa0662" # 从文件末尾提取的API密钥 +API_KEY = "sk-amQHwiEzPIZIB2KuF5A10dC23a0e4b02B48a7a2b6aFa0662" # 请求头 headers = { @@ -25,8 +25,16 @@ try: # 解析响应 result = response.json() - print("API响应:") - print(json.dumps(result, ensure_ascii=False, indent=2)) + + # 提取并打印返回的文本内容 + try: + # 从响应中获取文本内容 + text_content = result['choices'][0]['message']['content'] + print("\n提取的文本内容:") + print(text_content) + except (KeyError, IndexError) as e: + print(f"\n提取文本内容时出错: {e}") + print("响应结构可能与预期不符。") except requests.exceptions.RequestException as e: print(f"请求发生错误: {e}") \ No newline at end of file