ashish-ninehertz commited on
Commit
e0fb2f6
·
1 Parent(s): bbf9bf7
Files changed (2) hide show
  1. src/llm.py +1 -1
  2. src/main.py +2 -1
src/llm.py CHANGED
@@ -1,7 +1,7 @@
1
  import logging
2
  import ollama
3
  from typing import List, Dict, Optional
4
- from app.config import Config
5
  import os
6
 
7
 
 
1
  import logging
2
  import ollama
3
  from typing import List, Dict, Optional
4
+ from src.config import Config
5
  import os
6
 
7
 
src/main.py CHANGED
@@ -17,7 +17,6 @@ from sentence_transformers import SentenceTransformer
17
  from langchain_community.llms import Ollama
18
  import json
19
  import asyncio
20
- from src.llm import GeminiProvider
21
  from src.config import Config
22
  from qdrant_client.http.exceptions import UnexpectedResponse
23
 
@@ -58,6 +57,8 @@ class SessionManager:
58
  def __init__(self):
59
  """Initialize with in-memory sessions and Qdrant connection"""
60
  from src.prompts.templates import rag_prompt_template
 
 
61
  self.sessions = {} # In-memory session store
62
  self.embedding_model = SentenceTransformer(Config.EMBEDDING_MODEL)
63
  self.qdrant_client = QdrantClient(
 
17
  from langchain_community.llms import Ollama
18
  import json
19
  import asyncio
 
20
  from src.config import Config
21
  from qdrant_client.http.exceptions import UnexpectedResponse
22
 
 
57
  def __init__(self):
58
  """Initialize with in-memory sessions and Qdrant connection"""
59
  from src.prompts.templates import rag_prompt_template
60
+ from src.llm import GeminiProvider
61
+ self.llm = GeminiProvider()
62
  self.sessions = {} # In-memory session store
63
  self.embedding_model = SentenceTransformer(Config.EMBEDDING_MODEL)
64
  self.qdrant_client = QdrantClient(