1 line
17 KiB
JSON
1 line
17 KiB
JSON
{"repo": "aflybird0/haok", "files": {"haok/config/settings.py": {"ddb68fdff5077a4959220f555feb8b440228267e": {"t": 1785746915.1584036, "n": 1748, "h": "b840c52349b4fb8f", "c": "from functools import lru_cache\n\nfrom pydantic import BaseSettings, Field\n\n\nclass CustomBaseSettings(BaseSettings):\n class Config:\n env_file = \".env\"\n env_file_encoding = \"utf-8\"\n\n\n# class LLMSettings(CustomBaseSettings):\n# api_base: str = Field(env=\"OPENAI_API_BASE\", default=\"https://api.nextapi.fun/v1\")\n# api_key: str = Field(env=\"OPENAI_API_KEY\", default=\"ak-nyH5E1ka6oFKqRzxxqAooqcKSGCrk38mMgvU1DTqgOgRyTUQ\") # 论文临时用的key\n# default_model: str = Field(env=\"DEFAULT_MODEL\", default=\"gpt-3.5-turbo-0125\")\n # default_model: str = Field(env=\"DEFAULT_MODEL\", default=\"gpt-4-0125-preview\")\n\nclass LLMSettings(CustomBaseSettings):\n api_base: str = Field(env=\"OPENAI_API_BASE\", default=\"https://api.deepseek.com/v1\")\n api_key: str = Field(env=\"OPENAI_API_KEY\", default=\"sk-a0cb33389f42442a917c0dd978d962b2\") # 论文临时用的key\n default_model: str = Field(env=\"DEFAULT_MODEL\", default=\"deepseek-chat\")\n\n\nclass DatabaseSettings(CustomBaseSettings):\n url: str = Field(env=\"DATABASE_URL\", default=\"mongodb+srv://aflybird0:8ORG2lDRRm36ntP7@cluster0.lh2vpp8.mongodb.net/?retryWrites=true&w=majority\")\n db_name: str = Field(env=\"DATABASE_NAME\", default=\"cluster0\")\n\nclass DockerSettings(CustomBaseSettings):\n docker_port: str = Field(env=\"DOCKER_PORT\", default=\"8080\")\n docker_host: str = Field(env=\"DOCKER_HOST\", default=\"localhost\")\n\n\nclass Settings(BaseSettings):\n llm: LLMSettings = Field(default_factory=LLMSettings)\n database: DatabaseSettings = Field(default_factory=DatabaseSettings)\n docker: DockerSettings = Field(default_factory=DockerSettings)\n\n\n@lru_cache\ndef get_settings():\n settings = Settings()\n return settings\n\n\nif __name__ == '__main__':\n print(get_settings().json(indent=2))"}, "71b56ff212410e64dbf829b154023fda95a8b8c9": {"t": 1785768925.2138965, "n": 1748, "h": "b840c52349b4fb8f", "c": "from functools import lru_cache\n\nfrom pydantic import BaseSettings, Field\n\n\nclass CustomBaseSettings(BaseSettings):\n class Config:\n env_file = \".env\"\n env_file_encoding = \"utf-8\"\n\n\n# class LLMSettings(CustomBaseSettings):\n# api_base: str = Field(env=\"OPENAI_API_BASE\", default=\"https://api.nextapi.fun/v1\")\n# api_key: str = Field(env=\"OPENAI_API_KEY\", default=\"ak-nyH5E1ka6oFKqRzxxqAooqcKSGCrk38mMgvU1DTqgOgRyTUQ\") # 论文临时用的key\n# default_model: str = Field(env=\"DEFAULT_MODEL\", default=\"gpt-3.5-turbo-0125\")\n # default_model: str = Field(env=\"DEFAULT_MODEL\", default=\"gpt-4-0125-preview\")\n\nclass LLMSettings(CustomBaseSettings):\n api_base: str = Field(env=\"OPENAI_API_BASE\", default=\"https://api.deepseek.com/v1\")\n api_key: str = Field(env=\"OPENAI_API_KEY\", default=\"sk-a0cb33389f42442a917c0dd978d962b2\") # 论文临时用的key\n default_model: str = Field(env=\"DEFAULT_MODEL\", default=\"deepseek-chat\")\n\n\nclass DatabaseSettings(CustomBaseSettings):\n url: str = Field(env=\"DATABASE_URL\", default=\"mongodb+srv://aflybird0:8ORG2lDRRm36ntP7@cluster0.lh2vpp8.mongodb.net/?retryWrites=true&w=majority\")\n db_name: str = Field(env=\"DATABASE_NAME\", default=\"cluster0\")\n\nclass DockerSettings(CustomBaseSettings):\n docker_port: str = Field(env=\"DOCKER_PORT\", default=\"8080\")\n docker_host: str = Field(env=\"DOCKER_HOST\", default=\"localhost\")\n\n\nclass Settings(BaseSettings):\n llm: LLMSettings = Field(default_factory=LLMSettings)\n database: DatabaseSettings = Field(default_factory=DatabaseSettings)\n docker: DockerSettings = Field(default_factory=DockerSettings)\n\n\n@lru_cache\ndef get_settings():\n settings = Settings()\n return settings\n\n\nif __name__ == '__main__':\n print(get_settings().json(indent=2))"}, "edb123e1b389e6334dd962e59fe3a174d51d2c45": {"t": 1785768933.0973992, "n": 1698, "h": "897bf29f7d4387c7", "c": "from functools import lru_cache\n\nfrom pydantic import BaseSettings, Field\n\n\nclass CustomBaseSettings(BaseSettings):\n class Config:\n env_file = \".env\"\n env_file_encoding = \"utf-8\"\n\n\nclass LLMSettings(CustomBaseSettings):\n # api_base: str = Field(env=\"OPENAI_API_BASE\", default=\"https://api.gptapi.cyou/v1\")\n # api_key: str = Field(env=\"OPENAI_API_KEY\", default=\"sk-MrymAdMULNCmANziD6Bb5e51925047EdBf416377294bE673\")\n # default_model: str = Field(env=\"DEFAULT_MODEL\", default=\"gpt-3.5-turbo\")\n\n api_base: str = Field(env=\"OPENAI_API_BASE\", default=\"https://api.nextapi.fun/v1\")\n api_key: str = Field(env=\"OPENAI_API_KEY\", default=\"ak-rijxawD5sZxxnXXEoJmpWG52QlAMYxxaEQmp1ixrcJQOtbVY\")\n default_model: str = Field(env=\"DEFAULT_MODEL\", default=\"gpt-3.5-turbo-0125\")\n # default_model: str = Field(env=\"DEFAULT_MODEL\", default=\"gpt-4-0125-preview\")\n\n\nclass DatabaseSettings(CustomBaseSettings):\n url: str = Field(env=\"DATABASE_URL\", default=\"mongodb+srv://aflybird0:8ORG2lDRRm36ntP7@cluster0.lh2vpp8.mongodb.net/?retryWrites=true&w=majority\")\n db_name: str = Field(env=\"DATABASE_NAME\", default=\"cluster0\")\n\nclass DockerSettings(CustomBaseSettings):\n docker_port: str = Field(env=\"DOCKER_PORT\", default=\"8080\")\n docker_host: str = Field(env=\"DOCKER_HOST\", default=\"localhost\")\n\n\nclass Settings(BaseSettings):\n llm: LLMSettings = Field(default_factory=LLMSettings)\n database: DatabaseSettings = Field(default_factory=DatabaseSettings)\n docker: DockerSettings = Field(default_factory=DockerSettings)\n\n\n@lru_cache\ndef get_settings():\n settings = Settings()\n return settings\n\n\nif __name__ == '__main__':\n print(get_settings().json(indent=2))"}}, "haok/llm/openai.py": {"ddb68fdff5077a4959220f555feb8b440228267e": {"t": 1785747516.6362596, "n": 2853, "h": "bc325e16e6842b31", "c": "from langchain_openai import ChatOpenAI\nfrom haok.config.settings import get_settings\n\nopenai_api_key = get_settings().llm.api_key\nopenai_api_base = get_settings().llm.api_base\ndefault_model = get_settings().llm.default_model\n\nclass OpenAIConfig:\n\n openai_params = {\n \"openai_api_key\": openai_api_key,\n \"openai_api_base\": openai_api_base\n }\n\n @staticmethod\n def chatglm4():\n return ChatOpenAI(\n model_name=\"glm-4\",\n openai_api_base=\"https://open.bigmodel.cn/api/paas/v4\",\n # 论文临时用的key\n openai_api_key=\"eyJhbGciOiJIUzI1NiIsInNpZ25fdHlwZSI6IlNJR04iLCJ0eXAiOiJKV1QifQ.eyJhcGlfa2V5IjoiNjI5NzcxMjc5MzA3NTU4YzVlODI3MTdmYmUzMWIyZGMiLCJleHAiOjE3MTY3OTk4Nzk3ODYsInRpbWVzdGFtcCI6MTcwOTAyMzg3OTc4Nn0.83ZPAoa_ZC7dvvs7J7p_dtHP9eXc3sea8e2OOtszjWw\",\n streaming=False,\n )\n\n @staticmethod\n def openai():\n # os.environ[\"OPENAI_API_KEY\"] = OpenAIConfig.openai_api_key\n return ChatOpenAI(\n # model_name=\"gpt-3.5-turbo-1106\",\n model_name=default_model,\n # model_name=\"gpt-3.5-turbo\",\n openai_api_key=openai_api_key,\n openai_api_base=openai_api_base,\n temperature=0,\n )\n\n @staticmethod\n def openai4():\n # os.environ[\"OPENAI_API_KEY\"] = OpenAIConfig.openai_api_key\n return ChatOpenAI(\n # model_name=\"gpt-3.5-turbo-1106\",\n model_name=\"gpt-4-0125-preview\",\n # model_name=\"gpt-3.5-turbo\",\n openai_api_key=openai_api_key,\n openai_api_base=openai_api_base,\n temperature=0,\n )\n\n @staticmethod\n def openai_with_params(model: str=default_model, temperature: float=0.0):\n return ChatOpenAI(\n model_name=model,\n openai_api_key=openai_api_key,\n openai_api_base=openai_api_base,\n temperature=temperature,\n )\n\n @staticmethod\n def deep_seek_chat(temperature=0.0):\n return ChatOpenAI(\n model_name=\"deepseek-chat\",\n openai_api_key=\"sk-a0cb33389f42442a917c0dd978d962b2\", # 论文临时用的key\n openai_api_base=\"https://api.deepseek.com/v1\",\n temperature=temperature)\n\n @staticmethod\n def defaultLLM():\n # return OpenAIConfig.chatglm4()\n # return OpenAIConfig.openai()\n return OpenAIConfig.deep_seek_chat()\n\n @staticmethod\n def llm_with_params(model: str=default_model, temperature: float=0.0):\n # 如果model含有gpt\n if \"gpt\" in model:\n return ChatOpenAI(\n model_name=model,\n openai_api_key=openai_api_key,\n openai_api_base=openai_api_base,\n temperature=temperature,\n )\n elif \"glm\" in model:\n return OpenAIConfig.chatglm4()\n else:\n return OpenAIConfig.defaultLLM()"}, "71b56ff212410e64dbf829b154023fda95a8b8c9": {"t": 1785768891.8096561, "n": 2853, "h": "bc325e16e6842b31", "c": "from langchain_openai import ChatOpenAI\nfrom haok.config.settings import get_settings\n\nopenai_api_key = get_settings().llm.api_key\nopenai_api_base = get_settings().llm.api_base\ndefault_model = get_settings().llm.default_model\n\nclass OpenAIConfig:\n\n openai_params = {\n \"openai_api_key\": openai_api_key,\n \"openai_api_base\": openai_api_base\n }\n\n @staticmethod\n def chatglm4():\n return ChatOpenAI(\n model_name=\"glm-4\",\n openai_api_base=\"https://open.bigmodel.cn/api/paas/v4\",\n # 论文临时用的key\n openai_api_key=\"eyJhbGciOiJIUzI1NiIsInNpZ25fdHlwZSI6IlNJR04iLCJ0eXAiOiJKV1QifQ.eyJhcGlfa2V5IjoiNjI5NzcxMjc5MzA3NTU4YzVlODI3MTdmYmUzMWIyZGMiLCJleHAiOjE3MTY3OTk4Nzk3ODYsInRpbWVzdGFtcCI6MTcwOTAyMzg3OTc4Nn0.83ZPAoa_ZC7dvvs7J7p_dtHP9eXc3sea8e2OOtszjWw\",\n streaming=False,\n )\n\n @staticmethod\n def openai():\n # os.environ[\"OPENAI_API_KEY\"] = OpenAIConfig.openai_api_key\n return ChatOpenAI(\n # model_name=\"gpt-3.5-turbo-1106\",\n model_name=default_model,\n # model_name=\"gpt-3.5-turbo\",\n openai_api_key=openai_api_key,\n openai_api_base=openai_api_base,\n temperature=0,\n )\n\n @staticmethod\n def openai4():\n # os.environ[\"OPENAI_API_KEY\"] = OpenAIConfig.openai_api_key\n return ChatOpenAI(\n # model_name=\"gpt-3.5-turbo-1106\",\n model_name=\"gpt-4-0125-preview\",\n # model_name=\"gpt-3.5-turbo\",\n openai_api_key=openai_api_key,\n openai_api_base=openai_api_base,\n temperature=0,\n )\n\n @staticmethod\n def openai_with_params(model: str=default_model, temperature: float=0.0):\n return ChatOpenAI(\n model_name=model,\n openai_api_key=openai_api_key,\n openai_api_base=openai_api_base,\n temperature=temperature,\n )\n\n @staticmethod\n def deep_seek_chat(temperature=0.0):\n return ChatOpenAI(\n model_name=\"deepseek-chat\",\n openai_api_key=\"sk-a0cb33389f42442a917c0dd978d962b2\", # 论文临时用的key\n openai_api_base=\"https://api.deepseek.com/v1\",\n temperature=temperature)\n\n @staticmethod\n def defaultLLM():\n # return OpenAIConfig.chatglm4()\n # return OpenAIConfig.openai()\n return OpenAIConfig.deep_seek_chat()\n\n @staticmethod\n def llm_with_params(model: str=default_model, temperature: float=0.0):\n # 如果model含有gpt\n if \"gpt\" in model:\n return ChatOpenAI(\n model_name=model,\n openai_api_key=openai_api_key,\n openai_api_base=openai_api_base,\n temperature=temperature,\n )\n elif \"glm\" in model:\n return OpenAIConfig.chatglm4()\n else:\n return OpenAIConfig.defaultLLM()"}, "304d45d939a17f8f17afde679d09d6c915290bda": {"t": 1785768896.8220587, "n": 2483, "h": "68c000ccc6995f10", "c": "from langchain_openai import ChatOpenAI\nfrom haok.config.settings import get_settings\n\nopenai_api_key = get_settings().llm.api_key\nopenai_api_base = get_settings().llm.api_base\ndefault_model = get_settings().llm.default_model\n\nclass OpenAIConfig:\n\n openai_params = {\n \"openai_api_key\": openai_api_key,\n \"openai_api_base\": openai_api_base\n }\n\n @staticmethod\n def chatglm4():\n return ChatOpenAI(\n model_name=\"glm-4\",\n openai_api_base=\"https://open.bigmodel.cn/api/paas/v4\",\n openai_api_key=\"eyJhbGciOiJIUzI1NiIsInNpZ25fdHlwZSI6IlNJR04iLCJ0eXAiOiJKV1QifQ.eyJhcGlfa2V5IjoiNjI5NzcxMjc5MzA3NTU4YzVlODI3MTdmYmUzMWIyZGMiLCJleHAiOjE3MTY3OTk4Nzk3ODYsInRpbWVzdGFtcCI6MTcwOTAyMzg3OTc4Nn0.83ZPAoa_ZC7dvvs7J7p_dtHP9eXc3sea8e2OOtszjWw\",\n streaming=False,\n )\n\n @staticmethod\n def openai():\n # os.environ[\"OPENAI_API_KEY\"] = OpenAIConfig.openai_api_key\n return ChatOpenAI(\n # model_name=\"gpt-3.5-turbo-1106\",\n model_name=default_model,\n # model_name=\"gpt-3.5-turbo\",\n openai_api_key=openai_api_key,\n openai_api_base=openai_api_base,\n temperature=0,\n )\n\n @staticmethod\n def openai4():\n # os.environ[\"OPENAI_API_KEY\"] = OpenAIConfig.openai_api_key\n return ChatOpenAI(\n # model_name=\"gpt-3.5-turbo-1106\",\n model_name=\"gpt-4-0125-preview\",\n # model_name=\"gpt-3.5-turbo\",\n openai_api_key=openai_api_key,\n openai_api_base=openai_api_base,\n temperature=0,\n )\n\n @staticmethod\n def openai_with_params(model: str=default_model, temperature: float=0.0):\n return ChatOpenAI(\n model_name=model,\n openai_api_key=openai_api_key,\n openai_api_base=openai_api_base,\n temperature=temperature,\n )\n\n @staticmethod\n def defaultLLM():\n # return OpenAIConfig.chatglm4()\n return OpenAIConfig.openai()\n\n @staticmethod\n def llm_with_params(model: str=default_model, temperature: float=0.0):\n # 如果model含有gpt\n if \"gpt\" in model:\n return ChatOpenAI(\n model_name=model,\n openai_api_key=openai_api_key,\n openai_api_base=openai_api_base,\n temperature=temperature,\n )\n elif \"glm\" in model:\n return OpenAIConfig.chatglm4()\n else:\n return OpenAIConfig.defaultLLM()"}, "edb123e1b389e6334dd962e59fe3a174d51d2c45": {"t": 1785768899.4699836, "n": 2090, "h": "360254f669e906f0", "c": "from langchain_openai import ChatOpenAI\nfrom haok.config.settings import get_settings\n\nopenai_api_key = get_settings().llm.api_key\nopenai_api_base = get_settings().llm.api_base\ndefault_model = get_settings().llm.default_model\n\nclass OpenAIConfig:\n\n openai_params = {\n \"openai_api_key\": openai_api_key,\n \"openai_api_base\": openai_api_base\n }\n\n @staticmethod\n def chatglm4():\n return ChatOpenAI(\n model_name=\"glm-4\",\n openai_api_base=\"https://open.bigmodel.cn/api/paas/v4\",\n openai_api_key=\"eyJhbGciOiJIUzI1NiIsInNpZ25fdHlwZSI6IlNJR04iLCJ0eXAiOiJKV1QifQ.eyJhcGlfa2V5IjoiNjI5NzcxMjc5MzA3NTU4YzVlODI3MTdmYmUzMWIyZGMiLCJleHAiOjE3MTY3OTk4Nzk3ODYsInRpbWVzdGFtcCI6MTcwOTAyMzg3OTc4Nn0.83ZPAoa_ZC7dvvs7J7p_dtHP9eXc3sea8e2OOtszjWw\",\n streaming=False,\n )\n\n @staticmethod\n def openai():\n # os.environ[\"OPENAI_API_KEY\"] = OpenAIConfig.openai_api_key\n return ChatOpenAI(\n # model_name=\"gpt-3.5-turbo-1106\",\n model_name=default_model,\n # model_name=\"gpt-3.5-turbo\",\n openai_api_key=openai_api_key,\n openai_api_base=openai_api_base,\n temperature=0,\n )\n\n @staticmethod\n def openai_with_params(model: str=default_model, temperature: float=0.0):\n return ChatOpenAI(\n model_name=model,\n openai_api_key=openai_api_key,\n openai_api_base=openai_api_base,\n temperature=temperature,\n )\n\n @staticmethod\n def defaultLLM():\n # return OpenAIConfig.chatglm4()\n return OpenAIConfig.openai()\n\n @staticmethod\n def llm_with_params(model: str=default_model, temperature: float=0.0):\n # 如果model含有gpt\n if \"gpt\" in model:\n return ChatOpenAI(\n model_name=model,\n openai_api_key=openai_api_key,\n openai_api_base=openai_api_base,\n temperature=temperature,\n )\n elif \"glm\" in model:\n return OpenAIConfig.chatglm4()\n else:\n return OpenAIConfig.defaultLLM()"}}}} |