Merge pull request #1083 from iorisa/fixbug/0.7/issue/958

fixbug: #958
This commit is contained in:
Alexander Wu 2024-03-23 18:00:28 +08:00 committed by GitHub
commit 3bcf044f47
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -60,8 +60,8 @@ class GeminiLLM(BaseLLM):
def __init_gemini(self, config: LLMConfig):
if config.proxy:
logger.info(f"Use proxy: {config.proxy}")
os.environ["HTTP_PROXY"] = config.proxy
os.environ["HTTP_PROXYS"] = config.proxy
os.environ["http_proxy"] = config.proxy
os.environ["https_proxy"] = config.proxy
genai.configure(api_key=config.api_key)
def _user_msg(self, msg: str, images: Optional[Union[str, list[str]]] = None) -> dict[str, str]: