mirror of
https://github.com/QuivrHQ/quivr.git
synced 2024-12-15 01:21:48 +03:00
33 lines
1.1 KiB
Python
33 lines
1.1 KiB
Python
from typing import Annotated, Any, Dict, List, Tuple, Union
|
|
|
|
from fastapi import Depends
|
|
from langchain.embeddings.openai import OpenAIEmbeddings
|
|
from pydantic import BaseSettings
|
|
from supabase import Client, create_client
|
|
from vectorstore.supabase import SupabaseVectorStore
|
|
|
|
|
|
class BrainSettings(BaseSettings):
|
|
openai_api_key: str
|
|
anthropic_api_key: str
|
|
supabase_url: str
|
|
supabase_service_key: str
|
|
|
|
|
|
def common_dependencies() -> dict:
|
|
settings = BrainSettings()
|
|
embeddings = OpenAIEmbeddings(openai_api_key=settings.openai_api_key)
|
|
supabase_client: Client = create_client(settings.supabase_url, settings.supabase_service_key)
|
|
documents_vector_store = SupabaseVectorStore(
|
|
supabase_client, embeddings, table_name="vectors")
|
|
summaries_vector_store = SupabaseVectorStore(
|
|
supabase_client, embeddings, table_name="summaries")
|
|
|
|
return {
|
|
"supabase": supabase_client,
|
|
"embeddings": embeddings,
|
|
"documents_vector_store": documents_vector_store,
|
|
"summaries_vector_store": summaries_vector_store
|
|
}
|
|
|
|
CommonsDep = Annotated[dict, Depends(common_dependencies)] |