Suppression des modules contenant dotenv dans rag.py
Browse files- requirements.txt +0 -4
- src/rag.py +0 -9
requirements.txt
CHANGED
@@ -1,6 +1,2 @@
|
|
1 |
pypdf
|
2 |
-
openai
|
3 |
huggingface-hub
|
4 |
-
ollama
|
5 |
-
mistralai
|
6 |
-
dotenv
|
|
|
1 |
pypdf
|
|
|
2 |
huggingface-hub
|
|
|
|
|
|
src/rag.py
CHANGED
@@ -4,10 +4,7 @@ from pypdf import PdfReader
|
|
4 |
|
5 |
from .chunker import Chunker
|
6 |
from .amodel import ModelType
|
7 |
-
from .model_openai import OpenAIModel
|
8 |
from .model_huggingface import HuggingFaceModel
|
9 |
-
from .model_ollama import OllamaModel
|
10 |
-
from .model_mistral import MistralModel
|
11 |
from .store import Store
|
12 |
|
13 |
|
@@ -46,14 +43,8 @@ class Rag:
|
|
46 |
self.model_type = model_type
|
47 |
try:
|
48 |
match model_type:
|
49 |
-
case ModelType.MTOPENAI:
|
50 |
-
self.model = OpenAIModel("gpt-4o-mini", "text-embedding-3-small", 0)
|
51 |
case ModelType.MTHUGGINGFACE:
|
52 |
self.model = HuggingFaceModel("meta-llama/Meta-Llama-3-8B-Instruct", "sentence-transformers/all-MiniLM-l6-v2", 0)
|
53 |
-
case ModelType.MTOLLAMA:
|
54 |
-
self.model = OllamaModel("llama3.2:1b", "nomic-embed-text", 0.0)
|
55 |
-
case ModelType.MTMISTRAL:
|
56 |
-
self.model = MistralModel("mistral-large-latest", "mistral-embed", 0.0)
|
57 |
case _:
|
58 |
raise Exception("Rag.__init__: Unknown model type: {mt} : {v}".format(mt=ModelType.to_str(model_type), v=model_type))
|
59 |
self.emb_store = Store(store_dir) # persistant
|
|
|
4 |
|
5 |
from .chunker import Chunker
|
6 |
from .amodel import ModelType
|
|
|
7 |
from .model_huggingface import HuggingFaceModel
|
|
|
|
|
8 |
from .store import Store
|
9 |
|
10 |
|
|
|
43 |
self.model_type = model_type
|
44 |
try:
|
45 |
match model_type:
|
|
|
|
|
46 |
case ModelType.MTHUGGINGFACE:
|
47 |
self.model = HuggingFaceModel("meta-llama/Meta-Llama-3-8B-Instruct", "sentence-transformers/all-MiniLM-l6-v2", 0)
|
|
|
|
|
|
|
|
|
48 |
case _:
|
49 |
raise Exception("Rag.__init__: Unknown model type: {mt} : {v}".format(mt=ModelType.to_str(model_type), v=model_type))
|
50 |
self.emb_store = Store(store_dir) # persistant
|