ricoh51 commited on
Commit
941f36a
·
1 Parent(s): 698bfa5

Suppression des modules contenant dotenv dans rag.py

Browse files
Files changed (2) hide show
  1. requirements.txt +0 -4
  2. 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