jina rerank working
This commit is contained in:
39
test_ollama_embed_api.py
Normal file
39
test_ollama_embed_api.py
Normal file
@@ -0,0 +1,39 @@
|
||||
#!/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()
|
||||
Reference in New Issue
Block a user