From 9320a9c9722a3b65524a375958ce9a542f236a9f Mon Sep 17 00:00:00 2001 From: ayulockin Date: Mon, 13 May 2024 10:05:23 +0100 Subject: [PATCH] nits --- src/wandbot/chat/chat.py | 16 ++++++++-------- src/wandbot/ingestion/__main__.py | 10 ++++++---- src/wandbot/rag/query_handler.py | 2 +- src/wandbot/rag/response_synthesis.py | 2 +- 4 files changed, 16 insertions(+), 14 deletions(-) diff --git a/src/wandbot/chat/chat.py b/src/wandbot/chat/chat.py index 3faf7b3..cb66d8e 100644 --- a/src/wandbot/chat/chat.py +++ b/src/wandbot/chat/chat.py @@ -26,7 +26,7 @@ """ from typing import List -from weave.monitoring import StreamTable +# from weave.monitoring import StreamTable import wandb from wandbot.chat.config import ChatConfig @@ -66,11 +66,11 @@ def __init__( job_type="chat", ) self.run._label(repo="wandbot") - self.stream_table = StreamTable( - table_name="chat_logs", - project_name=self.config.wandb_project, - entity_name=self.config.wandb_entity, - ) + # self.stream_table = StreamTable( + # table_name="chat_logs", + # project_name=self.config.wandb_project, + # entity_name=self.config.wandb_entity, + # ) self.rag_pipeline = RAGPipeline(vector_store=vector_store) @@ -109,7 +109,7 @@ def __call__(self, chat_request: ChatRequest) -> ChatResponse: } result_dict.update({"application": chat_request.application}) self.run.log(usage_stats) - self.stream_table.log(result_dict) + # self.stream_table.log(result_dict) return ChatResponse(**result_dict) except Exception as e: with Timer() as timer: @@ -131,5 +131,5 @@ def __call__(self, chat_request: ChatRequest) -> ChatResponse: "end_time": timer.stop, } ) - self.stream_table.log(result) + # self.stream_table.log(result) return ChatResponse(**result) diff --git a/src/wandbot/ingestion/__main__.py b/src/wandbot/ingestion/__main__.py index a974346..704a628 100644 --- a/src/wandbot/ingestion/__main__.py +++ b/src/wandbot/ingestion/__main__.py @@ -11,13 +11,15 @@ def main(): project = os.environ.get("WANDB_PROJECT", "wandbot-dev") entity = os.environ.get("WANDB_ENTITY", "wandbot") - raw_artifact = prepare_data.load(project, entity) - preprocessed_artifact = preprocess_data.load(project, entity, raw_artifact) + # raw_artifact = prepare_data.load(project, entity) + raw_artifact = "wandbot/wandbot-dev/raw_dataset:v56" + # preprocessed_artifact = preprocess_data.load(project, entity, raw_artifact) + preprocessed_artifact = "wandbot/wandbot-dev/transformed_data:v23" vectorstore_artifact = vectorstores.load( - project, entity, preprocessed_artifact + project, entity, preprocessed_artifact, "chroma_index" ) - create_ingestion_report(project, entity, raw_artifact, vectorstore_artifact) + # create_ingestion_report(project, entity, raw_artifact, vectorstore_artifact) print(vectorstore_artifact) diff --git a/src/wandbot/rag/query_handler.py b/src/wandbot/rag/query_handler.py index 857c9fb..cdde16b 100644 --- a/src/wandbot/rag/query_handler.py +++ b/src/wandbot/rag/query_handler.py @@ -262,7 +262,7 @@ class QueryEnhancer: def __init__( self, model: str = "gpt-4-0125-preview", - fallback_model: str = "gpt-4-1106-preview", + fallback_model: str = "gpt-4-0125-preview", ): self.model = model # type: ignore self.fallback_model = fallback_model # type: ignore diff --git a/src/wandbot/rag/response_synthesis.py b/src/wandbot/rag/response_synthesis.py index 22729d7..65d9ac1 100644 --- a/src/wandbot/rag/response_synthesis.py +++ b/src/wandbot/rag/response_synthesis.py @@ -118,7 +118,7 @@ class ResponseSynthesizer: def __init__( self, model: str = "gpt-4-0125-preview", - fallback_model: str = "gpt-4-1106-preview", + fallback_model: str = "gpt-4-0125-preview", ): self.model = model # type: ignore self.fallback_model = fallback_model # type: ignore