refactor(brainsettings): added moved files

This commit is contained in:
Stan Girard 2023-06-19 21:21:26 +02:00
parent c222f57b22
commit b8ec6fd342
3 changed files with 2 additions and 8 deletions

View File

@ -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:

View File

@ -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__)

View File

@ -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