StanGirard/quivr
1
1
Fork 0
mirror of https://github.com/StanGirard/quivr.git synced 2024-09-11 13:15:41 +03:00
Code Issues Packages Projects Releases Wiki Activity

type common_dependencies

Browse Source
This commit is contained in:
Chloe Mouret 2023-08-01 12:31:19 +02:00
parent 1558300997
commit 38ead5f61f
1 changed files with 9 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
backend/core/models/settings.py
View File

@ -1,4 +1,4 @@
from typing import Annotated
from typing import Annotated, TypedDict
from fastapi import Depends
from langchain.embeddings.openai import OpenAIEmbeddings
@ -26,7 +26,14 @@ class LLMSettings(BaseSettings):
model_path: str = "./local_models/ggml-gpt4all-j-v1.3-groovy.bin"
def common_dependencies() -> dict:
class CommonDependencies(TypedDict):
supabase: Client
embeddings: OpenAIEmbeddings
documents_vector_store: SupabaseVectorStore
summaries_vector_store: SupabaseVectorStore
def common_dependencies() -> CommonDependencies:
settings = BrainSettings() # pyright: ignore reportPrivateUsage=none
embeddings = OpenAIEmbeddings(
openai_api_key=settings.openai_api_key