Singtel_Use_Case1 / quick_test.py
cosmoruler
stuck already
db6dcad
raw
history blame contribute delete
531 Bytes
#!/usr/bin/env python3
"""
Quick test of Ollama with a simple prompt
"""
import ollama
def quick_test():
print("πŸ” Quick Ollama test...")
try:
# Very simple test
response = ollama.generate(
model='llama2',
prompt='Say "Hello" in one word only.'
)
print(f"βœ… Response: {response['response']}")
return True
except Exception as e:
print(f"❌ Failed: {e}")
return False
if __name__ == "__main__":
quick_test()