概覽
LangChain 是一個用於構建 LLM 應用程式的熱門框架。LemonData 可與 LangChain 的 OpenAI 整合無縫協作。安裝
複製
pip install langchain langchain-openai
基礎配置
複製
from langchain_openai import ChatOpenAI
llm = ChatOpenAI(
model="gpt-4o",
api_key="sk-your-lemondata-key",
base_url="https://api.lemondata.cc/v1"
)
response = llm.invoke("Hello, how are you?")
print(response.content)
使用不同的模型
存取任何 LemonData 模型:複製
# OpenAI GPT-4o
gpt4 = ChatOpenAI(
model="gpt-4o",
api_key="sk-your-key",
base_url="https://api.lemondata.cc/v1"
)
# Anthropic Claude
claude = ChatOpenAI(
model="claude-sonnet-4-5",
api_key="sk-your-key",
base_url="https://api.lemondata.cc/v1"
)
# Google Gemini
gemini = ChatOpenAI(
model="gemini-2.5-flash",
api_key="sk-your-key",
base_url="https://api.lemondata.cc/v1"
)
# DeepSeek
deepseek = ChatOpenAI(
model="deepseek-r1",
api_key="sk-your-key",
base_url="https://api.lemondata.cc/v1"
)
使用訊息紀錄進行對話
複製
from langchain_core.messages import HumanMessage, SystemMessage
messages = [
SystemMessage(content="You are a helpful assistant."),
HumanMessage(content="What is the capital of France?")
]
response = llm.invoke(messages)
print(response.content)
串流 (Streaming)
複製
for chunk in llm.stream("Write a poem about coding"):
print(chunk.content, end="", flush=True)
非同步用法
複製
import asyncio
async def main():
response = await llm.ainvoke("Hello!")
print(response.content)
asyncio.run(main())
鏈 (Chains)
複製
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.output_parsers import StrOutputParser
prompt = ChatPromptTemplate.from_messages([
("system", "You are a helpful assistant that translates {input_language} to {output_language}."),
("human", "{text}")
])
chain = prompt | llm | StrOutputParser()
result = chain.invoke({
"input_language": "English",
"output_language": "French",
"text": "Hello, how are you?"
})
print(result)
RAG (檢索增強生成)
複製
from langchain_openai import OpenAIEmbeddings
from langchain_community.vectorstores import FAISS
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.runnables import RunnablePassthrough
# Embeddings
embeddings = OpenAIEmbeddings(
model="text-embedding-3-small",
api_key="sk-your-key",
base_url="https://api.lemondata.cc/v1"
)
# Create vector store
texts = ["LemonData supports 300+ AI models", "API is OpenAI compatible"]
vectorstore = FAISS.from_texts(texts, embeddings)
retriever = vectorstore.as_retriever()
# RAG chain
template = """Answer based on context:
{context}
Question: {question}
"""
prompt = ChatPromptTemplate.from_template(template)
rag_chain = (
{"context": retriever, "question": RunnablePassthrough()}
| prompt
| llm
)
response = rag_chain.invoke("How many models does LemonData support?")
print(response.content)
代理 (Agents)
LangChain 中的 agent API 正在不斷演進。對於新專案,請考慮使用 LangGraph 以獲得更靈活的代理架構。
複製
from langchain.agents import create_openai_tools_agent, AgentExecutor
from langchain_core.prompts import ChatPromptTemplate
from langchain_core.tools import tool
@tool
def search(query: str) -> str:
"""Search for information."""
return f"Search results for: {query}"
tools = [search]
prompt = ChatPromptTemplate.from_messages([
("system", "You are a helpful assistant with access to tools."),
("human", "{input}"),
("placeholder", "{agent_scratchpad}")
])
agent = create_openai_tools_agent(llm, tools, prompt)
executor = AgentExecutor(agent=agent, tools=tools)
result = executor.invoke({"input": "Search for LemonData pricing"})
print(result["output"])
環境變數
為了使程式碼更簡潔,請使用環境變數:複製
export OPENAI_API_KEY="sk-your-lemondata-key"
export OPENAI_API_BASE="https://api.lemondata.cc/v1"
複製
from langchain_openai import ChatOpenAI
# Will automatically use environment variables
llm = ChatOpenAI(model="gpt-4o")
回呼 (Callbacks) 與追蹤
複製
from langchain_core.callbacks import StdOutCallbackHandler
llm = ChatOpenAI(
model="gpt-4o",
api_key="sk-your-key",
base_url="https://api.lemondata.cc/v1",
callbacks=[StdOutCallbackHandler()]
)
最佳實踐
根據成本選擇合適的模型
根據成本選擇合適的模型
在鏈中使用較便宜的模型 (GPT-4o-mini) 處理簡單任務。
實作重試機制
實作重試機制
LangChain 針對暫時性錯誤內建了重試邏輯。
監控 token 使用量
監控 token 使用量
使用回呼 (callbacks) 來追蹤 token 消耗。