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)