Langchain 学习02 ， basic example02

最新推荐文章于 2024-07-25 22:24:14 发布

leichangqing

最新推荐文章于 2024-07-25 22:24:14 发布

阅读量357

点赞数 8

分类专栏： Langchain 文章标签： langchain 学习

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

版权

Langchain 专栏收录该内容

6 篇文章 0 订阅

订阅专栏

本文介绍了如何使用Langchain库与AzureOpenAI的GPT-35-turbo模型进行交互，通过提供关于icecream的提示，获取相关的笑话输出。

摘要由CSDN通过智能技术生成

Langchain 学习02 ， basic example

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_core.output_parsers import StrOutputParser
from langchain_core.prompts import ChatPromptTemplate
from langchain_openai import AzureOpenAI

input = {"topic": "ice cream"}
prompt = ChatPromptTemplate.from_template("tell me a short joke about {topic}")
prompt_value = prompt.invoke(input)

llm = AzureOpenAI(
    model=model,
    deployment_name=deployment_name,
    api_key=api_key,
    azure_endpoint=azure_endpoint,
    api_version=api_version,
)

message = llm.invoke(prompt_value)
output_parser = StrOutputParser()
output_parser.invoke(message)
## or 下面写法
# input = {"topic": "ice cream"}
# prompt.invoke(input)
# (prompt | model).invoke(input)