lizhilun/AI_NovelGenerator
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

🎨 gemini timeout params

Browse Source
This commit is contained in:
osi
2025-05-13 16:25:25 +08:00
parent c8a9fe87f6
commit f15b00d159
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
llm_adapters.py
+3 -3
View File
@@ -103,9 +103,10 @@ class GeminiAdapter(BaseLLMAdapter):
self.model_name = model_name self.model_name = model_name
self.max_tokens = max_tokens self.max_tokens = max_tokens
self.temperature = temperature self.temperature = temperature
self.timeout = timeout # gemini超时时间是毫秒
self.timeout = timeout * 1000
self._client = genai.Client(api_key=self.api_key,http_options=types.HttpOptions(base_url=base_url)) self._client = genai.Client(api_key=self.api_key,http_options=types.HttpOptions(base_url=base_url,timeout=self.timeout))
def invoke(self, prompt: str) -> str: def invoke(self, prompt: str) -> str:
try: try:
@@ -116,7 +117,6 @@ class GeminiAdapter(BaseLLMAdapter):
max_output_tokens=self.max_tokens, max_output_tokens=self.max_tokens,
temperature=self.temperature, temperature=self.temperature,
), ),
# timeout=self.timeout # 添加超时参数
) )
if response and response.text: if response and response.text:
return response.text return response.text