|
|
|
|
@ -77,15 +77,10 @@ class OllamaEmbeddingModel(TextEmbeddingModel):
|
|
|
|
|
inputs.append(text)
|
|
|
|
|
|
|
|
|
|
# Prepare the payload for the request
|
|
|
|
|
payload = {
|
|
|
|
|
"input": inputs,
|
|
|
|
|
"model": model,
|
|
|
|
|
}
|
|
|
|
|
payload = {"input": inputs, "model": model, "options": {"use_mmap": "true"}}
|
|
|
|
|
|
|
|
|
|
# Make the request to the OpenAI API
|
|
|
|
|
response = requests.post(
|
|
|
|
|
endpoint_url, headers=headers, data=json.dumps(payload), timeout=(10, 300), options={"use_mmap": "true"}
|
|
|
|
|
)
|
|
|
|
|
# Make the request to the Ollama API
|
|
|
|
|
response = requests.post(endpoint_url, headers=headers, data=json.dumps(payload), timeout=(10, 300))
|
|
|
|
|
|
|
|
|
|
response.raise_for_status() # Raise an exception for HTTP errors
|
|
|
|
|
response_data = response.json()
|
|
|
|
|
|