From 6dfdde94ac053ffdbcf1a72d2c825be5fa4f163e Mon Sep 17 00:00:00 2001 From: lihengda Date: Tue, 14 May 2024 19:30:02 +0800 Subject: [PATCH] =?UTF-8?q?=E5=9F=BA=E4=BA=8E=20ChatGLM2-6B=20=E5=AE=9E?= =?UTF-8?q?=E7=8E=B0=E5=B8=A6=E5=9B=BE=E5=BD=A2=E5=8C=96=E7=95=8C=E9=9D=A2?= =?UTF-8?q?=E7=9A=84=20openai-translator?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .idea/.gitignore | 8 ++++++++ .../ai_translator/translator/translation_chain.py | 4 ++++ 2 files changed, 12 insertions(+) create mode 100644 .idea/.gitignore diff --git a/.idea/.gitignore b/.idea/.gitignore new file mode 100644 index 00000000..13566b81 --- /dev/null +++ b/.idea/.gitignore @@ -0,0 +1,8 @@ +# Default ignored files +/shelf/ +/workspace.xml +# Editor-based HTTP Client requests +/httpRequests/ +# Datasource local storage ignored files +/dataSources/ +/dataSources.local.xml diff --git a/langchain/openai-translator/ai_translator/translator/translation_chain.py b/langchain/openai-translator/ai_translator/translator/translation_chain.py index dbcf53fc..d2082722 100644 --- a/langchain/openai-translator/ai_translator/translator/translation_chain.py +++ b/langchain/openai-translator/ai_translator/translator/translation_chain.py @@ -1,5 +1,6 @@ from langchain_openai import ChatOpenAI from langchain.chains import LLMChain +from langchain.llms import ChatGLM from langchain.prompts.chat import ( ChatPromptTemplate, @@ -30,6 +31,9 @@ def __init__(self, model_name: str = "gpt-3.5-turbo", verbose: bool = True): # 为了翻译结果的稳定性，将 temperature 设置为 0 chat = ChatOpenAI(model_name=model_name, temperature=0, verbose=verbose) + if model_name == "chat_glm": + endpoint_url = ("http://127.0.0.1:8000") # endpoint_url 填写跑模型的地址 + chat = ChatGLM(endpoint_url=endpoint_url, temperature=0, verbose=verbose) self.chain = LLMChain(llm=chat, prompt=chat_prompt_template, verbose=verbose)