Migrate parts to library

Author: vmesel

src/dialog/models/db.py renamed to src/dialog/db/__init__.py

@@ -15,7 +15,7 @@
 from dialog_lib.db.memory import generate_memory_instance
 from dialog.llm.embeddings import EMBEDDINGS_LLM
 from dialog.settings import Settings
-from dialog.models.db import get_session
+from dialog.db import get_session
 
 class DialogLLM(AbstractLLM):
     def __init__(self, *args, **kwargs):

src/dialog/llm/default.py

@@ -1,12 +1,5 @@
-from typing import List
-
 from langchain_openai import OpenAIEmbeddings
-from sqlalchemy import select
 
-from dialog.models import CompanyContent
-from dialog.models.db import get_session
 from dialog.settings import Settings
-from dialog_lib.embeddings import generate
-
 
 EMBEDDINGS_LLM = OpenAIEmbeddings(openai_api_key=Settings().OPENAI_API_KEY)

src/dialog/llm/embeddings.py

@@ -4,7 +4,7 @@
 from decouple import config as d_config
 from sqlalchemy import engine_from_config, pool
 
-from dialog.models.db import Base
+from dialog_lib.db.models import Base
 from pgvector.sqlalchemy import Vector
 from logging.config import fileConfig
 

src/dialog/migrations/env.py

@@ -5,9 +5,9 @@
 
 from dialog.llm import get_llm_class
 from dialog_lib.db.memory import get_messages
-from dialog.models import Chat as ChatEntity
+from dialog_lib.db.models import Chat as ChatEntity
 from dialog.schemas import ChatModel, SessionModel, SessionsModel
-from dialog.models.db import engine, get_session
+from dialog.db import engine, get_session
 from dialog.settings import Settings
 
 from sqlalchemy import text

src/dialog/models/__init__.py

@@ -6,8 +6,8 @@
 
 from dialog_lib.embeddings.generate import generate_embeddings
 from dialog.llm.embeddings import EMBEDDINGS_LLM
-from dialog.models import CompanyContent
-from dialog.models.db import get_session
+from dialog_lib.db.models import CompanyContent
+from dialog.db import get_session
 
 session = next(get_session())
 

src/dialog/models/helpers.py

@@ -2,12 +2,12 @@
 import pytest
 
 from main import app
-from dialog.models.db import Base
+from dialog_lib.db.models import Base
 from fastapi.testclient import TestClient
 from sqlalchemy import create_engine
 from sqlalchemy.orm import sessionmaker
 from dialog_lib.db.utils import create_chat_session
-from dialog.models.db import get_session
+from dialog.db import get_session
 
 SQLALCHEMY_DATABASE_URL = "postgresql://talkdai:talkdai@db/test_talkdai"
 

src/dialog/routers/dialog.py

@@ -1,6 +1,6 @@
 import pytest
 
-from dialog.models import ChatMessages, Chat
+from dialog_lib.db.models import ChatMessages, Chat
 
 def test_health(client):
     response = client.get("/health")