Local SE Agent Masterclass

import ollama
from typing import List, Dict, Any, Optional

class OllamaLLMClient:
    """
    Wrapper สำหรับควบคุมการสื่อสารกับโมเดลภาษาในเครื่องผ่าน Ollama Daemon
    """
    def __init__(self, model_name: str = "qwen2.5-coder:7b", host: Optional[str] = None):
        self.model_name = model_name
        # ถ้าระบุ host จะชี้ไป endpoint ที่กำหนด เช่น http://localhost:11434
        self.client = ollama.Client(host=host) if host else ollama.Client()

    def check_health(self) -> bool:
        """ตรวจสอบว่า daemon พร้อมใช้งานและมีโมเดลเป้าหมายอยู่หรือไม่"""
        try:
            models_response = self.client.list()
            # ดึงรายชื่อโมเดลทั้งหมดที่มีอยู่ในเครื่อง
            available_models = [m.model for m in models_response.models]
            # ตรวจว่าโมเดลเป้าหมายมีอยู่ในเครื่องหรือไม่ (เช่น qwen2.5-coder:7b หรือ qwen2.5-coder:7b-latest)
            return any(self.model_name in name for name in available_models)
        except Exception:
            return False

    def chat(self, messages: List[Dict[str, str]], temperature: float = 0.2) -> str:
        """
        ส่งประวัติบทสนทนาและรับข้อความตอบกลับจากโมเดล
        :param messages: ลิสต์ของข้อความในรูปแบบ [{'role': 'user', 'content': '...'}]
        :param temperature: ค่าความสุ่ม ยิ่งต่ำยิ่งคงเส้นคงวาและเป็นเหตุเป็นผล (เหมาะกับงานโค้ด)
        """
        try:
            response = self.client.chat(
                model=self.model_name,
                messages=messages,
                options={"temperature": temperature}
            )
            # ดึงเฉพาะข้อความคำตอบที่โมเดลส่งออกมา
            return response.message.content
        except ollama.ResponseError as e:
            raise RuntimeError(f"Ollama Response Error (Status {e.status_code}): {e.error}")
        except Exception as e:
            raise ConnectionError(f"ไม่สามารถเชื่อมต่อ Ollama daemon ที่ localhost:11434 ได้: {str(e)}")

from core.client import OllamaLLMClient

def main():
    print("🔍 กำลังตรวจสอบการเชื่อมต่อกับ Ollama Daemon บน Mac M3...")
    client = OllamaLLMClient(model_name="qwen2.5-coder:7b")
    
    if not client.check_health():
        print("❌ ข้อผิดพลาด: ไม่พบ Ollama Daemon หรือยังไม่ได้ดึงโมเดล 'qwen2.5-coder:7b'")
        print("👉 กรุณารันคำสั่ง: ollama run qwen2.5-coder:7b ใน Terminal อีกหน้าต่างหนึ่ง")
        return

    print("✅ การเชื่อมต่อสมบูรณ์! กำลังทดสอบส่ง Prompt ไปประมวลผล...")
    messages = [
        {"role": "system", "content": "You are a concise SE Assistant."},
        {"role": "user", "content": "เขียนฟังก์ชัน Python สั้นๆ สำหรับกลับข้อความ (reverse string) 1 บรรทัด"}
    ]
    
    answer = client.chat(messages, temperature=0.1)
    print("\n--- ผลลัพธ์จากโมเดล ---")
    print(answer)
    print("------------------------")
    print("🎉 ยินดีด้วยครับ! ท่อส่งข้อมูล Day 1 ทำงานได้สมบูรณ์แบบ 100%")

if __name__ == "__main__":
    main()