evolution-prometheus / rag_system.py
WladlenVK's picture
Create rag_system.py
e8a21b2 unverified
Raw History Blame Contribute Delete
382 Bytes
import os
from langchain_openai import OpenAIEmbeddings, ChatOpenAI
from langchain_community.vectorstores import PGVector
class BookRAG:
def __init__(self):
# Настройки будут подгружаться из секретного файла .env
self.embeddings = OpenAIEmbeddings()
self.llm = ChatOpenAI(model="gpt-4o-mini", temperature=0.1)