简单的langchain使用模版

最新推荐文章于 2025-12-11 11:44:25 发布

原创

最新推荐文章于 2025-12-11 11:44:25 发布 · 244 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#langchain #python #开发语言

from langchain.chains import SimpleChain
from langchain.prompts import PromptTemplate
from langchain.llms import OpenAI
from langchain.callbacks import CallbackManager
from langchain.callbacks.stdout import StdOutCallbackHandler
from langchain.memory import SimpleMemory

# 定义预处理函数，用于清理输入文本
def preprocess_text(text):
    return text.strip().lower()

# 定义后处理函数，用于处理模型输出
def postprocess_text(text):
    return text.capitalize()

# 定义一个简单的错误处理函数
def error_handler(e):
    print(f"Error encountered: {e}")
    return "Sorry, an error occurred."

# 定义一个提示模板
prompt_template = PromptTemplate(
    input_variables=["input_text"],
    template="将以下文本翻译成中文: {input_text}"
)

# 创建一个OpenAI模型实例（假设你已经配置了API密钥）
llm = OpenAI(model="text-davinci-003")

# 使用CallbackManager来管理回调
callback_manager = CallbackManager([StdOutCallbackHandler()])

# 使用SimpleMemory来管理会话记忆
memory = SimpleMemory()

# 创建一个完善的链式处理流程
class EnhancedChain(SimpleChain):
    def __init__(self, prompt_template,