feat: gemini + proxy

This commit is contained in:
莘权 马 2024-03-07 12:27:28 +08:00
parent 26252e68bd
commit f6260ec084
2 changed files with 6 additions and 2 deletions

View file

@ -1,7 +1,7 @@
#!/usr/bin/env python
# -*- coding: utf-8 -*-
# @Desc : Google Gemini LLM from https://ai.google.dev/tutorials/python_quickstart
import os
from typing import Optional, Union
import google.generativeai as genai
@ -58,6 +58,10 @@ class GeminiLLM(BaseLLM):
self.llm = GeminiGenerativeModel(model_name=self.model)
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
genai.configure(api_key=config.api_key)
def _user_msg(self, msg: str, images: Optional[Union[str, list[str]]] = None) -> dict[str, str]:

View file

@ -57,7 +57,7 @@ extras_require["dev"] = (["pylint~=3.0.3", "black~=23.3.0", "isort~=5.12.0", "pr
setup(
name="metagpt",
version="0.7.4",
version="0.7.5",
description="The Multi-Agent Framework",
long_description=long_description,
long_description_content_type="text/markdown",