Langchain 学习04 LCEL 01

leichangqing

于 2024-03-07 21:18:55 发布

阅读量360

点赞数 8

分类专栏： Langchain 文章标签： langchain flask

本文链接：https://blog.csdn.net/leichangqing/article/details/136546391

版权

Langchain 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

Langchain 学习04 LCEL 01

import os
from dotenv import load_dotenv, find_dotenv
_ = load_dotenv(find_dotenv())  # 读取本地 .env 文件，里面定义了 OPENAI_API_KEY

#############
api_key = os.environ['AZURE_OPENAI_API_KEY']
azure_endpoint = os.environ['AZURE_OPENAI_ENDPOINT']
api_version = os.environ['AZURE_OPENAPI_VERSION']
model="gpt-35-turbo"
deployment_name="gpt-35-turbo"
############
from langchain_openai import AzureChatOpenAI
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.output_parsers import StrOutputParser
from langchain_core.runnables import RunnablePassthrough


prompt = ChatPromptTemplate.from_template(
    "Tell me a short joke about {topic}"
)
output_parser = StrOutputParser()
model = AzureChatOpenAI(
    model=model,
    deployment_name=deployment_name,
    api_key=api_key,
    azure_endpoint=azure_endpoint,
    api_version=api_version,
    )

chain = (
    {"topic": RunnablePassthrough()}
    | prompt
    | model
    | output_parser
)

print(chain.invoke("ice cream"))