はじめに
前のレッスンで、Ollama API についてはすでに理解しました。シンプルなチャットボットから RAG パイプラインまで、すべて Mac 上でローカルに実行できる実際の AI アプリを Python で構築しましょう。
1. Ollama Python ライブラリ
インストール
pip3 install ollama
簡単なチャット
import ollama
response = ollama.chat(
model='llama3.2',
messages=[
{'role': 'system', 'content': 'Trả lời bằng tiếng Việt, ngắn gọn.'},
{'role': 'user', 'content': 'FastAPI vs Flask: nên chọn cái nào?'},
]
)
print(response['message']['content'])
ストリーミング
import ollama
stream = ollama.chat(
model='llama3.2',
messages=[{'role': 'user', 'content': 'Viết function merge sort bằng Python'}],
stream=True,
)
for chunk in stream:
print(chunk['message']['content'], end='', flush=True)
print()
2. メモリ付きチャットボット端末
#!/usr/bin/env python3
"""Chatbot terminal chạy local với Ollama."""
import ollama
MODEL = "llama3.2"
SYSTEM_PROMPT = """Bạn là xDev AI, trợ lý lập trình thông minh.
- Trả lời bằng tiếng Việt
- Ngắn gọn, đi thẳng vào vấn đề
- Viết code khi được hỏi
- Giải thích rõ ràng cho beginner"""
def main():
messages = [{"role": "system", "content": SYSTEM_PROMPT}]
print("🤖 xDev AI Chatbot (local)")
print(f"📦 Model: {MODEL}")
print("💡 Gõ 'quit' để thoát, 'clear' để xóa history\n")
while True:
try:
user_input = input("👤 You: ").strip()
except (KeyboardInterrupt, EOFError):
print("\n\n👋 Tạm biệt!")
break
if not user_input:
continue
if user_input.lower() == 'quit':
print("👋 Tạm biệt!")
break
if user_input.lower() == 'clear':
messages = [{"role": "system", "content": SYSTEM_PROMPT}]
print("🗑️ Đã xóa lịch sử chat\n")
continue
messages.append({"role": "user", "content": user_input})
print("🤖 AI: ", end="", flush=True)
full_response = ""
stream = ollama.chat(model=MODEL, messages=messages, stream=True)
for chunk in stream:
token = chunk['message']['content']
print(token, end="", flush=True)
full_response += token
print("\n")
messages.append({"role": "assistant", "content": full_response})
# Giới hạn history để không tốn quá nhiều context
if len(messages) > 21: # system + 10 turns
messages = [messages[0]] + messages[-20:]
if __name__ == "__main__":
main()
タインを保存 chatbot.py そして実行します:
python3 chatbot.py
3. リッチなUIを備えたチャットボット(端末)
ライブラリを使用して UI をアップグレードする rich:
pip3 install rich
#!/usr/bin/env python3
"""Chatbot terminal với Rich UI."""
import ollama
from rich.console import Console
from rich.markdown import Markdown
from rich.panel import Panel
from rich.prompt import Prompt
console = Console()
MODEL = "llama3.2"
def main():
messages = [
{"role": "system", "content": "Trả lời bằng tiếng Việt. Dùng markdown formatting."}
]
console.print(Panel.fit(
"🤖 [bold cyan]xDev AI Chatbot[/]\n"
f"📦 Model: [green]{MODEL}[/] (local)\n"
"💡 Gõ [yellow]quit[/] để thoát",
border_style="cyan"
))
while True:
user_input = Prompt.ask("\n[bold green]You[/]")
if user_input.lower() in ('quit', 'exit'):
console.print("\n[dim]👋 Bye![/]")
break
messages.append({"role": "user", "content": user_input})
# Collect full response
full_response = ""
with console.status("[cyan]Đang suy nghĩ...[/]"):
response = ollama.chat(model=MODEL, messages=messages)
full_response = response['message']['content']
# Render markdown
console.print()
console.print(Panel(
Markdown(full_response),
title="🤖 AI",
border_style="blue",
padding=(1, 2)
))
messages.append({"role": "assistant", "content": full_response})
if __name__ == "__main__":
main()
4. ラングチェーン + オラマ
LangChain は、LLM アプリケーションの最も人気のあるフレームワークです。 Ollama をネイティブでサポートします。
インストール
pip3 install langchain langchain-ollama
簡単なチャット
from langchain_ollama import ChatOllama
llm = ChatOllama(model="llama3.2", temperature=0.7)
response = llm.invoke("Docker là gì? Trả lời bằng tiếng Việt, 3 câu.")
print(response.content)
プロンプトテンプレートを使用したチェーン
from langchain_ollama import ChatOllama
from langchain_core.prompts import ChatPromptTemplate
llm = ChatOllama(model="llama3.2")
prompt = ChatPromptTemplate.from_messages([
("system", "Bạn là chuyên gia {topic}. Trả lời bằng tiếng Việt."),
("user", "{question}"),
])
chain = prompt | llm
result = chain.invoke({
"topic": "DevOps",
"question": "CI/CD pipeline cơ bản gồm những bước gì?"
})
print(result.content)
5. ローカルの埋め込み
埋め込みはテキストをベクトルに変換します。これはセマンティック検索と RAG に不可欠です。
Ollama API を使用する
import ollama
# Pull embedding model (chỉ cần lần đầu)
# ollama pull nomic-embed-text
response = ollama.embeddings(
model='nomic-embed-text',
prompt='Docker là công nghệ containerization'
)
embedding = response['embedding']
print(f"Dimension: {len(embedding)}") # 768
print(f"First 5: {embedding[:5]}")
類似度を計算する
import ollama
import numpy as np
def get_embedding(text):
response = ollama.embeddings(model='nomic-embed-text', prompt=text)
return np.array(response['embedding'])
def cosine_similarity(a, b):
return np.dot(a, b) / (np.linalg.norm(a) * np.linalg.norm(b))
# So sánh semantic similarity
texts = [
"Docker container giúp đóng gói ứng dụng",
"Kubernetes orchestrates containerized applications",
"Python là ngôn ngữ lập trình phổ biến",
"Cách deploy app với Docker Compose",
]
query = "Làm sao chạy ứng dụng trong container?"
query_emb = get_embedding(query)
print(f"Query: {query}\n")
for text in texts:
text_emb = get_embedding(text)
sim = cosine_similarity(query_emb, text_emb)
print(f" {sim:.4f} | {text}")
出力:
Query: Làm sao chạy ứng dụng trong container?
0.8234 | Docker container giúp đóng gói ứng dụng
0.7891 | Kubernetes orchestrates containerized applications
0.4123 | Python là ngôn ngữ lập trình phổ biến
0.8456 | Cách deploy app với Docker Compose
6. 単純な RAG (検索拡張生成)
RAG = ドキュメントから関連情報を取得 → プロンプトに入力 → LLM がより正確に回答します。
#!/usr/bin/env python3
"""Simple RAG with Ollama - hoàn toàn local."""
import ollama
import numpy as np
# --- Bước 1: Chuẩn bị knowledge base ---
documents = [
"FastAPI là web framework Python hiện đại, hỗ trợ async/await và tự động tạo OpenAPI docs.",
"Docker Compose dùng file YAML để define và chạy multi-container Docker applications.",
"PostgreSQL là hệ quản trị CSDL quan hệ mã nguồn mở, hỗ trợ JSON, full-text search.",
"Kubernetes (K8s) là hệ thống orchestration container, tự động scale và heal applications.",
"Redis là in-memory data store, thường dùng làm cache, message broker, session store.",
"Nginx là web server và reverse proxy phổ biến, dùng để load balancing và serving static files.",
"Git là hệ thống version control phân tán, GitHub là platform hosting Git repositories.",
"CI/CD là quy trình tự động build, test và deploy code. Tools phổ biến: GitHub Actions, Jenkins.",
]
# --- Bước 2: Tạo embeddings cho documents ---
def get_embedding(text):
resp = ollama.embeddings(model='nomic-embed-text', prompt=text)
return np.array(resp['embedding'])
print("📚 Đang tạo embeddings cho knowledge base...")
doc_embeddings = [get_embedding(doc) for doc in documents]
print(f"✅ Đã index {len(documents)} documents\n")
# --- Bước 3: Retrieval ---
def retrieve(query, top_k=3):
query_emb = get_embedding(query)
similarities = [
np.dot(query_emb, doc_emb) / (np.linalg.norm(query_emb) * np.linalg.norm(doc_emb))
for doc_emb in doc_embeddings
]
top_indices = np.argsort(similarities)[-top_k:][::-1]
return [(documents[i], similarities[i]) for i in top_indices]
# --- Bước 4: Generate with context ---
def rag_query(question):
# Retrieve relevant documents
relevant = retrieve(question, top_k=3)
context = "\n".join([f"- {doc}" for doc, _ in relevant])
prompt = f"""Dựa vào thông tin sau đây để trả lời câu hỏi. Nếu thông tin không đủ, hãy nói rõ.
Thông tin:
{context}
Câu hỏi: {question}
Trả lời bằng tiếng Việt, ngắn gọn:"""
response = ollama.chat(
model='llama3.2',
messages=[{'role': 'user', 'content': prompt}]
)
print(f"📖 Retrieved documents (similarity):")
for doc, sim in relevant:
print(f" [{sim:.3f}] {doc[:80]}...")
print()
return response['message']['content']
# --- Sử dụng ---
questions = [
"FastAPI có hỗ trợ async không?",
"Dùng gì để cache trong Python app?",
"Làm sao tự động deploy code?",
]
for q in questions:
print(f"❓ {q}")
answer = rag_query(q)
print(f"💡 {answer}\n")
print("---\n")
7. 非同期のサポート
import asyncio
import ollama
async def async_chat():
client = ollama.AsyncClient()
response = await client.chat(
model='llama3.2',
messages=[{'role': 'user', 'content': 'Hello!'}]
)
print(response['message']['content'])
# Streaming
stream = await client.chat(
model='llama3.2',
messages=[{'role': 'user', 'content': 'Viết haiku về code'}],
stream=True,
)
async for chunk in stream:
print(chunk['message']['content'], end='', flush=True)
asyncio.run(async_chat())
概要
| アプローチ | ツール | 使用例 |
|---|---|---|
| 簡単なチャット | ollama 図書館 | 高速スクリプト、シンプルなチャットボット |
| リッチなUI | ollama + rich | 美しいチャットボット端末 |
| ラングチェーン | langchain-ollama | チェーン、エージェント、複雑なパイプライン |
| 埋め込み | nomic-embed-text | セマンティック検索、RAG |
| 非同期 | ollama.AsyncClient | Web サーバー、同時リクエスト |
演習
1.基本的なチャットボットターミナルを実行し、10回チャットし、動作メモリを確認します
2. 履歴をJSONファイルに保存/読み込みする機能を追加
3. 興味のあるトピックに関する 20 以上のドキュメントを含むミニ RAG を構築します
4. LangChainを使って英語翻訳→要約→箇条書きのチェーンを作成する
5. (おまけ) ファイルを受信するための CLI ツールを作成する .py 内部のコードを説明します
次の記事: ビジョン モデル — ローカル画像の分析 →