mirror of
https://github.com/onyx-dot-app/onyx.git
synced 2026-10-23 11:53:08 -05:00
hagen-danswer
b97cc01bb2
* Added confluence permission syncing * seperated out group and doc syncing * minorbugfix and mypy * added frontend and fixed bug * Minor refactor * dealth with confluence rate limits! * mypy fixes!!! * addressed yuhong feedback * primary key fix
30 lines
1.0 KiB
Python
30 lines
1.0 KiB
Python
from sqlalchemy.orm import Session
|
|
|
|
from danswer.db.search_settings import get_current_search_settings
|
|
from danswer.document_index.interfaces import DocumentIndex
|
|
from danswer.document_index.vespa.index import VespaIndex
|
|
|
|
|
|
def get_default_document_index(
|
|
primary_index_name: str,
|
|
secondary_index_name: str | None,
|
|
) -> DocumentIndex:
|
|
"""Primary index is the index that is used for querying/updating etc.
|
|
Secondary index is for when both the currently used index and the upcoming
|
|
index both need to be updated, updates are applied to both indices"""
|
|
# Currently only supporting Vespa
|
|
return VespaIndex(
|
|
index_name=primary_index_name, secondary_index_name=secondary_index_name
|
|
)
|
|
|
|
|
|
def get_current_primary_default_document_index(db_session: Session) -> DocumentIndex:
|
|
"""
|
|
TODO: Use redis to cache this or something
|
|
"""
|
|
search_settings = get_current_search_settings(db_session)
|
|
return get_default_document_index(
|
|
primary_index_name=search_settings.index_name,
|
|
secondary_index_name=None,
|
|
)
|