87ed1c4425
- 升级Agent(chat_module=agent切换):升级到langgraph react agent逻辑、集成到主分支fay中、基于自动决策工具调用机制、基于日程跟踪的主动沟通、支持外部观测数据传入; - 修复因线程同步问题导致的配置文件读写不稳定 - 聊天采纳功能的bug修复
44 lines
1007 B
Python
44 lines
1007 B
Python
from fastapi import FastAPI
|
|
from langchain_core.prompts import ChatPromptTemplate
|
|
from langchain_core.output_parsers import StrOutputParser
|
|
from langchain_openai import ChatOpenAI
|
|
from langserve import add_routes
|
|
import os
|
|
os.environ["OPENAI_API_KEY"] = "sk-"
|
|
os.environ["OPENAI_API_BASE"] = "https://cn.api.zyai.online/v1"
|
|
|
|
# 1. Create prompt template
|
|
system_template = "Translate the following into {language}:"
|
|
prompt_template = ChatPromptTemplate.from_messages([
|
|
('system', system_template),
|
|
('user', '{text}')
|
|
])
|
|
|
|
# 2. Create model
|
|
model = ChatOpenAI()
|
|
|
|
# 3. Create parser
|
|
parser = StrOutputParser()
|
|
|
|
# 4. Create chain
|
|
chain = prompt_template | model | parser
|
|
|
|
|
|
# 4. App definition
|
|
app = FastAPI(
|
|
title="LangChain Server",
|
|
version="1.0",
|
|
description="A simple API server using LangChain's Runnable interfaces",
|
|
)
|
|
|
|
# 5. Adding chain route
|
|
add_routes(
|
|
app,
|
|
chain,
|
|
path="/chain",
|
|
)
|
|
|
|
if __name__ == "__main__":
|
|
import uvicorn
|
|
|
|
uvicorn.run(app, host="localhost", port=8000) |