Create ollama_with_python.py

2024-12-18 04:58:51 +08:00
parent 5be00bb720
commit ebf86b30eb
1 changed files with 10 additions and 0 deletions
--- a/2024.12.11_ollama/ollama_with_python.py
+++ b/2024.12.11_ollama/ollama_with_python.py
@@ -0,0 +1,10 @@
+# 直接输出
+import ollama
+response = ollama.chat(model="llama3.2:latest", messages=[{"role": "user","content": "你好"}], stream=False)
+print(response['message']['content'])
+
+# 流式输出
+import ollama
+response = ollama.chat(model="llama3.2:latest", messages=[{"role": "user", "content": "你好"}], stream=True)
+for part in response:
+    print(part['message']['content'], end='', flush=True)