prawnikai / config.py
adowu's picture
Update config.py
82d91f6 verified
raw
history blame
601 Bytes
import os
from dotenv import load_dotenv
load_dotenv()
# Konfiguracja API
HF_TOKEN = os.getenv('HF_TOKEN')
MODEL_NAME = "Qwen/Qwen2.5-72B-Instruct"
# Konfiguracja bazy danych
CHUNK_SIZE = 1000
CHUNK_OVERLAP = 200
DATABASE_DIR = "chroma_db"
# Konfiguracja modelu embeddings
EMBEDDING_MODEL = "sentence-transformers/paraphrase-multilingual-mpnet-base-v2"
# System prompt
SYSTEM_PROMPT = """Jesteś asystentem prawniczym specjalizującym się w polskim prawie.
Twoje odpowiedzi opierają się na aktualnych przepisach prawnych.
Zawsze cytuj konkretne artykuły i paragrafy z odpowiednich ustaw.
"""