39 lines
1.2 KiB
Python
39 lines
1.2 KiB
Python
#!/usr/bin/env python3
|
|
"""Test Ollama embedding API to understand format for reranking"""
|
|
|
|
import requests
|
|
import json
|
|
|
|
def test_ollama_embed():
|
|
print("=== Testing Ollama Embedding API ===")
|
|
|
|
# Test embedding with Jina rerank model
|
|
test_data = {
|
|
"model": "jina-reranker-v2:latest",
|
|
"input": ["The capital of France is Paris.", "Tokyo is the capital of Japan."]
|
|
}
|
|
|
|
try:
|
|
response = requests.post(
|
|
"http://localhost:11434/api/embed",
|
|
json=test_data,
|
|
timeout=10
|
|
)
|
|
|
|
print(f"Status: {response.status_code}")
|
|
if response.status_code == 200:
|
|
result = response.json()
|
|
print(f"Response keys: {list(result.keys())}")
|
|
print(f"Model: {result.get('model')}")
|
|
print(f"Embeddings length: {len(result.get('embeddings', []))}")
|
|
if result.get('embeddings'):
|
|
print(f"First embedding shape: {len(result['embeddings'][0])}")
|
|
print(f"First embedding sample: {result['embeddings'][0][:5]}...")
|
|
else:
|
|
print(f"Error: {response.text}")
|
|
|
|
except Exception as e:
|
|
print(f"Error: {e}")
|
|
|
|
if __name__ == "__main__":
|
|
test_ollama_embed() |