StanGirard/quivr
1
1
Fork 0
mirror of https://github.com/StanGirard/quivr.git synced 2024-11-23 21:22:35 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix: add missing properties (#530)

Browse Source
This commit is contained in:
Matt 2023-07-06 09:52:47 +01:00 committed by GitHub
parent 7e7a3113a5
commit f352005dcf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 22 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
backend/llm/openai_functions.py
View File

@ -1,12 +1,14 @@
from typing import Any, Dict, List, Optional from typing import Any, Dict, List, Optional
from langchain.chat_models import ChatOpenAI from langchain.chat_models import ChatOpenAI
from langchain.embeddings.openai import OpenAIEmbeddings
from llm.models.FunctionCall import FunctionCall from llm.models.FunctionCall import FunctionCall
from llm.models.OpenAiAnswer import OpenAiAnswer from llm.models.OpenAiAnswer import OpenAiAnswer
from logger import get_logger from logger import get_logger
from models.chat import ChatHistory from models.chat import ChatHistory
from repository.chat.get_chat_history import get_chat_history from repository.chat.get_chat_history import get_chat_history
from repository.chat.update_chat_history import update_chat_history from repository.chat.update_chat_history import update_chat_history
from supabase import Client, create_client
from vectorstore.supabase import CustomSupabaseVectorStore from vectorstore.supabase import CustomSupabaseVectorStore
from .base import BaseBrainPicking from .base import BaseBrainPicking
@ -61,6 +63,25 @@ class OpenAIFunctionsBrainPicking(BaseBrainPicking):
def openai_client(self) -> ChatOpenAI: def openai_client(self) -> ChatOpenAI:
return ChatOpenAI(openai_api_key=self.openai_api_key) return ChatOpenAI(openai_api_key=self.openai_api_key)
@property
def embeddings(self) -> OpenAIEmbeddings:
return OpenAIEmbeddings(openai_api_key=self.openai_api_key)
@property
def supabase_client(self) -> Client:
return create_client(
self.brain_settings.supabase_url, self.brain_settings.supabase_service_key
)
@property
def vector_store(self) -> CustomSupabaseVectorStore:
return CustomSupabaseVectorStore(
self.supabase_client,
self.embeddings,
table_name="vectors",
brain_id=self.brain_id,
)
def _get_model_response( def _get_model_response(
self, self,
messages: List[Dict[str, str]], messages: List[Dict[str, str]],
@ -103,14 +124,8 @@ class OpenAIFunctionsBrainPicking(BaseBrainPicking):
Retrieve documents related to the question Retrieve documents related to the question
""" """
logger.info("Getting context") logger.info("Getting context")
vector_store = CustomSupabaseVectorStore(
self.supabase_client,
self.embeddings,
table_name="vectors",
brain_id=self.brain_id,
)
return vector_store.similarity_search(query=question) return self.vector_store.similarity_search(query=question)
def _construct_prompt( def _construct_prompt(
self, question: str, useContext: bool = False, useHistory: bool = False self, question: str, useContext: bool = False, useHistory: bool = False