Add: python API method to add custom embeddings#1615
Conversation
|
Important Review skippedDraft detected. Please check the settings in the CodeRabbit UI or the ⚙️ Run configurationConfiguration used: Repository UI Review profile: CHILL Plan: Pro Run ID: You can disable this status message by setting the Use the checkbox below for a quick retry:
✨ Finishing Touches🧪 Generate unit tests (beta)
Thanks for using CodeRabbit! It's free for OSS, and your support helps us grow. If you like it, consider giving us a shout-out. Comment |
| db_migrations.run_migrations(engine=engine, engine_url=engine_url) | ||
|
|
||
|
|
||
| def _migrate_duckdb_schema(engine: Engine) -> None: |
There was a problem hiding this comment.
This we could also leave away
|
|
||
| def collection_has_embeddings(session: Session, collection_id: UUID) -> bool: | ||
| """Check if there are any embeddings available for the given collection. | ||
| class CollectionEmbeddingsStatus(SQLModel): |
There was a problem hiding this comment.
This was refactored. The new status tells us whether we have embeddings and whether we have text embeddings for search. We can later allow the user to hook up their own text encoder model to do also text based search with custom models.
What has changed and why?
How has it been tested?
Did you update CHANGELOG.md?