refactor: configure grader

config.yml

@@ -11,4 +11,8 @@ models:
   rag:
     model: cohere/command-r
     temperature: 0
-  
+  
+embeddings:
+  intfloat:
+    model: intfloat/multilingual-e5-large
+    device: cpu

lib/graph.py

@@ -14,15 +14,20 @@ from lib.runnables import (
     rag_chain,
     message_classificator,
 )
+from lib.config import Config
 from langgraph.graph import END, StateGraph
 from transformers import AutoModel, AutoTokenizer
 import torch
 import torch.nn.functional as F
 
-model_name = "intfloat/multilingual-e5-large"
+config = Config("config.yml")
+model_name = config.get("embeddings", "intfloat", "model")
+device = config.get("embeddings", "intfloat", "device")
+
 tokenizer = AutoTokenizer.from_pretrained(model_name)
 model = AutoModel.from_pretrained(model_name)
-device = torch.device("cuda")
+
+device = torch.device(device)
 model.to(device)
 SIMILARITY_TRESHHOLD = 0.8