Update app.py
Browse files
app.py
CHANGED
@@ -89,15 +89,31 @@ def load_custom_css():
|
|
89 |
""", unsafe_allow_html=True)
|
90 |
|
91 |
def initialize_pipeline():
|
92 |
-
"""Initialize RAG pipeline with configurations"""
|
93 |
try:
|
94 |
openai_api_key = os.getenv('OPENAI_API_KEY') or st.secrets['OPENAI_API_KEY']
|
95 |
-
|
96 |
-
|
97 |
-
config
|
98 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
99 |
pipeline = AcademicCalendarRAG(config)
|
100 |
|
|
|
101 |
with open("calendar.json", "r", encoding="utf-8") as f:
|
102 |
calendar_data = json.load(f)
|
103 |
pipeline.load_data(calendar_data)
|
|
|
89 |
""", unsafe_allow_html=True)
|
90 |
|
91 |
def initialize_pipeline():
|
92 |
+
"""Initialize RAG pipeline with enhanced configurations"""
|
93 |
try:
|
94 |
openai_api_key = os.getenv('OPENAI_API_KEY') or st.secrets['OPENAI_API_KEY']
|
95 |
+
|
96 |
+
# Create detailed configuration
|
97 |
+
config = PipelineConfig(
|
98 |
+
model=ModelConfig(
|
99 |
+
openai_api_key=openai_api_key,
|
100 |
+
embedder_model="sentence-transformers/paraphrase-multilingual-mpnet-base-v2",
|
101 |
+
openai_model="gpt-4",
|
102 |
+
temperature=0.3
|
103 |
+
),
|
104 |
+
retriever=RetrieverConfig(top_k=5),
|
105 |
+
cache=CacheConfig(
|
106 |
+
enabled=True,
|
107 |
+
cache_dir=Path("./cache"),
|
108 |
+
ttl=86400
|
109 |
+
),
|
110 |
+
processing=ProcessingConfig(batch_size=32),
|
111 |
+
localization=LocalizationConfig(enable_thai_normalization=True)
|
112 |
+
)
|
113 |
+
|
114 |
pipeline = AcademicCalendarRAG(config)
|
115 |
|
116 |
+
# Load calendar data
|
117 |
with open("calendar.json", "r", encoding="utf-8") as f:
|
118 |
calendar_data = json.load(f)
|
119 |
pipeline.load_data(calendar_data)
|