Spaces:
Sleeping
Sleeping
ashish-ninehertz
commited on
Commit
·
e0fb2f6
1
Parent(s):
bbf9bf7
changes
Browse files- src/llm.py +1 -1
- 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
|
| 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(
|