Merge pull request #72 from wandb/fix/v1.3/dependencies

Fix: Update v1.2 dependencies

pyproject.toml

@@ -27,7 +27,6 @@ discord = "^2.3.2"
 markdown = "^3.5.1"
 fastapi = "^0.109.2"
 tree-sitter-languages = "^1.7.1"
-cohere = "^4.32"
 markdownify = "^0.11.6"
 uvicorn = "^0.24.0"
 zenpy = "^2.0.46"
@@ -43,11 +42,18 @@ langchain = "^0.1.5"
 langchain-openai = "^0.0.5"
 chromadb = "^0.4.22"
 langchain-experimental = "^0.0.50"
-simsimd = "^3.7.4"
+simsimd = "3.7.7"
+langchain-core = "^0.1.44"
+langchain-cohere = "^0.1.3"
 
 [tool.poetry.dev-dependencies]
 #fasttext = {git = "https://github.com/cfculhane/fastText"} # FastText doesn't come with pybind11 and we need to use this workaround.
 
+[tool.poetry.group.eval.dependencies]
+ragas = "^0.1.7"
+dataclasses-json = "^0.6.4"
+llama-index = "^0.10.30"
+
 [build-system]
 requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"

run.sh

@@ -1,4 +1,4 @@
 (uvicorn wandbot.api.app:app --host="0.0.0.0" --port=8000) & \
 (python -m wandbot.apps.slack -l en) & \
 (python -m wandbot.apps.slack -l ja) & \
-(python -m wandbot.apps.discord) & \
+(python -m wandbot.apps.discord)

src/wandbot/api/app.py

@@ -33,8 +33,9 @@
 from datetime import datetime, timezone
 
 import pandas as pd
-import wandb
 from fastapi import FastAPI
+
+import wandb
 from wandbot.api.routers import chat as chat_router
 from wandbot.api.routers import database as database_router
 from wandbot.api.routers import retrieve as retrieve_router
@@ -110,4 +111,4 @@ async def backup_db():
 if __name__ == "__main__":
     import uvicorn
 
-    uvicorn.run(app, host="localhost", port=8000)
+    uvicorn.run(app, host="localhost", port=8000)

src/wandbot/chat/chat.py

@@ -26,14 +26,15 @@
 """
 from typing import List
 
+from weave.monitoring import StreamTable
+
 import wandb
 from wandbot.chat.config import ChatConfig
 from wandbot.chat.rag import RAGPipeline, RAGPipelineOutput
 from wandbot.chat.schemas import ChatRequest, ChatResponse
 from wandbot.database.schemas import QuestionAnswer
 from wandbot.retriever import VectorStore
 from wandbot.utils import Timer, get_logger
-from weave.monitoring import StreamTable
 
 logger = get_logger(__name__)
 

src/wandbot/chat/rag.py

@@ -3,6 +3,7 @@
 
 from langchain_community.callbacks import get_openai_callback
 from pydantic import BaseModel
+
 from wandbot.rag.query_handler import QueryEnhancer
 from wandbot.rag.response_synthesis import ResponseSynthesizer
 from wandbot.rag.retrieval import FusionRetrieval