Upload ollama-example.py

ollama-example.py

@@ -0,0 +1,74 @@
+import requests
+import json
+import time
+from datetime import datetime
+
+def send_request_to_ollama(prompt, model="TaiwanPro-rs"):
+    """向Ollama發送請求並獲取回應"""
+    url = "http://localhost:11434/api/generate"
+    
+    data = {
+        "model": model,
+        "prompt": prompt,
+        "stream": False
+    }
+    
+    try:
+        response = requests.post(url, json=data)
+        response.raise_for_status()
+        return response.json()["response"]
+    except requests.exceptions.RequestException as e:
+        print(f"請求錯誤: {e}")
+        return f"錯誤: {str(e)}"
+
+def initialize_markdown_file():
+    """初始化Markdown文件並寫入頭部信息"""
+    timestamp = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    
+    with open("output.md", "w", encoding="utf-8") as file:
+        file.write(f"# Ollama TaiwanPro-r 請求結果\n\n")
+        file.write(f"生成時間: {timestamp}\n\n")
+        file.write(f"模型: TaiwanPro-r\n\n")
+    
+    print(f"已初始化 output.md 文件")
+
+def append_to_markdown(index, prompt, response):
+    """將單個請求和回應添加到Markdown文件"""
+    with open("output.md", "a", encoding="utf-8") as file:
+        file.write(f"## 請求 {index}\n\n")
+        file.write(f"### 提示\n\n```\n{prompt}\n```\n\n")
+        file.write(f"### 回應\n\n{response}\n\n---\n\n")
+    
+    print(f"已將請求 {index} 的結果寫入 output.md")
+
+def main():
+    # 準備五則請求
+    requests = [
+        "介紹台灣的夜市文化",
+        "寫一首關於台北101的現代詩",
+        "請說明台灣的健保系統如何運作",
+        "分析台灣科技產業的發展歷程",
+        "台灣的原住民族有哪些，他們的文化特色是什麼？"
+    ]
+    
+    # 初始化markdown文件
+    initialize_markdown_file()
+    
+    print("開始向Ollama的TaiwanPro-r模型發送請求...")
+    
+    for i, prompt in enumerate(requests, 1):
+        print(f"發送請求 {i}/{len(requests)}: {prompt[:30]}...")
+        response = send_request_to_ollama(prompt)
+        print(f"已收到請求 {i} 的回應")
+        
+        # 立即將結果添加到文件中
+        append_to_markdown(i, prompt, response)
+        
+        # 避免過快發送請求
+        if i < len(requests):
+            time.sleep(1)
+    
+    print("所有請求已完成，所有結果已保存至 output.md")
+
+if __name__ == "__main__":
+    main()