Merge branch 'dev' into refactor/update-web-parsing
This commit is contained in:
commit
6c9b3d6385
2 changed files with 2 additions and 3 deletions
|
|
@ -124,7 +124,7 @@ class OllamaEmbeddingEngine(EmbeddingEngine):
|
||||||
self.endpoint, json=payload, headers=headers, timeout=60.0
|
self.endpoint, json=payload, headers=headers, timeout=60.0
|
||||||
) as response:
|
) as response:
|
||||||
data = await response.json()
|
data = await response.json()
|
||||||
return data["embedding"]
|
return data["embeddings"][0]
|
||||||
|
|
||||||
def get_vector_size(self) -> int:
|
def get_vector_size(self) -> int:
|
||||||
"""
|
"""
|
||||||
|
|
|
||||||
|
|
@ -324,7 +324,6 @@ class LanceDBAdapter(VectorDBInterface):
|
||||||
|
|
||||||
def get_data_point_schema(self, model_type: BaseModel):
|
def get_data_point_schema(self, model_type: BaseModel):
|
||||||
related_models_fields = []
|
related_models_fields = []
|
||||||
|
|
||||||
for field_name, field_config in model_type.model_fields.items():
|
for field_name, field_config in model_type.model_fields.items():
|
||||||
if hasattr(field_config, "model_fields"):
|
if hasattr(field_config, "model_fields"):
|
||||||
related_models_fields.append(field_name)
|
related_models_fields.append(field_name)
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue