mirror of
https://github.com/onyx-dot-app/onyx.git
synced 2026-10-23 03:42:59 -05:00
Chris Weaver
132a9f750d
Also fixes import ordering (previously, local imports weren't grouped together as they should have been)
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
import typesense # type: ignore
|
|
from qdrant_client import QdrantClient
|
|
|
|
from danswer.configs.app_configs import QDRANT_API_KEY
|
|
from danswer.configs.app_configs import QDRANT_HOST
|
|
from danswer.configs.app_configs import QDRANT_PORT
|
|
from danswer.configs.app_configs import QDRANT_URL
|
|
from danswer.configs.app_configs import TYPESENSE_API_KEY
|
|
from danswer.configs.app_configs import TYPESENSE_HOST
|
|
from danswer.configs.app_configs import TYPESENSE_PORT
|
|
|
|
|
|
_qdrant_client: QdrantClient | None = None
|
|
_typesense_client: typesense.Client | None = None
|
|
|
|
|
|
def get_qdrant_client() -> QdrantClient:
|
|
global _qdrant_client
|
|
if _qdrant_client is None:
|
|
if QDRANT_URL and QDRANT_API_KEY:
|
|
_qdrant_client = QdrantClient(url=QDRANT_URL, api_key=QDRANT_API_KEY)
|
|
elif QDRANT_HOST and QDRANT_PORT:
|
|
_qdrant_client = QdrantClient(host=QDRANT_HOST, port=QDRANT_PORT)
|
|
else:
|
|
raise Exception("Unable to instantiate QdrantClient")
|
|
|
|
return _qdrant_client
|
|
|
|
|
|
def get_typesense_client() -> typesense.Client:
|
|
global _typesense_client
|
|
if _typesense_client is None:
|
|
if TYPESENSE_HOST and TYPESENSE_PORT and TYPESENSE_API_KEY:
|
|
_typesense_client = typesense.Client(
|
|
{
|
|
"api_key": TYPESENSE_API_KEY,
|
|
"nodes": [
|
|
{
|
|
"host": TYPESENSE_HOST,
|
|
"port": str(TYPESENSE_PORT),
|
|
"protocol": "http",
|
|
}
|
|
],
|
|
}
|
|
)
|
|
else:
|
|
raise Exception("Unable to instantiate TypesenseClient")
|
|
|
|
return _typesense_client
|