fixbug: proxy

This commit is contained in:
莘权 马 2024-03-23 17:57:10 +08:00
parent 283c7b49c3
commit a988d9c4f3

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]: