一個簡單的LangChain Hello World範例,本篇以Anthropic Claude API為例。
環境
Windows 11。
事前要求
Anthropic Claude Code取得API key。
建立.env
在專案根目錄下新增一個.env檔,用來設定ANTHROPIC_API_KEY讓LangChain程式執行時可讀取。
ANTHROPIC_API_KEY="your_anthropic_api_key"
範例
在專案根目錄下的main.py撰寫以下內容,如同在語言模型的網頁版發送一次提示詞。
main.py
from langchain.chat_models import init_chat_model
from dotenv import load_dotenv
load_dotenv() # 載入.env設定
model = init_chat_model("claude-sonnet-4-6") # 建立一個語言模型物件
response = model.invoke("say 'hello world' only") # 給模型的提示詞
print(response.content) # 印出模型回應訊息
LangChain調用模型API需要API Key,所以要用load_dotenv()函式載入.env中的設定。
使用LangChain的init_chat_model函式,傳入指定的模型名稱,取得模型物件(BaseChatModel)。
使用模型物件的invoke並傳入給模型的提示詞(prompt),並返回模型的回應訊息物件(BaseMessage)。
最後印出回應物件的content為模型的回應文字訊息。
測試
在專案根目錄下cmd執行uv run main.py,結果如下。
C:\..\langchain-demo>uv run main.py
hello world
沒有留言:
張貼留言