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

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

View file

@ -56,8 +56,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]: