mirror of
https://github.com/StanGirard/quivr.git
synced 2024-09-11 13:15:41 +03:00
refactor(brainsettings): added moved files
This commit is contained in:
parent
c222f57b22
commit
b8ec6fd342
@ -17,18 +17,12 @@ from langchain.vectorstores import SupabaseVectorStore
|
||||
from llm.prompt import LANGUAGE_PROMPT
|
||||
from llm.prompt.CONDENSE_PROMPT import CONDENSE_QUESTION_PROMPT
|
||||
from models.chats import ChatMessage
|
||||
from models.settings import BrainSettings
|
||||
from pydantic import BaseModel, BaseSettings
|
||||
from supabase import Client, create_client
|
||||
from vectorstore.supabase import CustomSupabaseVectorStore
|
||||
|
||||
|
||||
class BrainSettings(BaseSettings):
|
||||
openai_api_key: str
|
||||
anthropic_api_key: str
|
||||
supabase_url: str
|
||||
supabase_service_key: str
|
||||
|
||||
|
||||
class AnswerConversationBufferMemory(ConversationBufferMemory):
|
||||
"""ref https://github.com/hwchase17/langchain/issues/5630#issuecomment-1574222564"""
|
||||
def save_context(self, inputs: Dict[str, Any], outputs: Dict[str, str]) -> None:
|
||||
|
@ -5,7 +5,6 @@ from fastapi import Depends
|
||||
from langchain.embeddings.openai import OpenAIEmbeddings
|
||||
from langchain.vectorstores import SupabaseVectorStore
|
||||
from logger import get_logger
|
||||
|
||||
from supabase import Client, create_client
|
||||
|
||||
logger = get_logger(__name__)
|
||||
|
@ -4,6 +4,7 @@ from llm.qa import BrainPicking, BrainSettings
|
||||
from llm.summarization import llm_evaluate_summaries, llm_summerize
|
||||
from logger import get_logger
|
||||
from models.chats import ChatMessage
|
||||
from models.settings import BrainSettings
|
||||
from pydantic import BaseModel
|
||||
from utils.common import CommonsDep
|
||||
|
||||
|
Loading…
Reference in New Issue
Block a user