Spaces:
Sleeping
Sleeping
Update routes.py
Browse files
routes.py
CHANGED
@@ -99,24 +99,6 @@ def send_query(query_request: QueryRequest):
|
|
99 |
Processes a query and returns the bot's response along with any web sources.
|
100 |
"""
|
101 |
try:
|
102 |
-
prompt_type = query_request.prompt_type.lower()
|
103 |
-
prompt_template_mapping = {
|
104 |
-
"university": PromptTemplates.get_university_chatbot_prompt,
|
105 |
-
"quiz_solving": PromptTemplates.get_quiz_solving_prompt,
|
106 |
-
"assignment_solving": PromptTemplates.get_assignment_solving_prompt,
|
107 |
-
"paper_solving": PromptTemplates.get_paper_solving_prompt,
|
108 |
-
"quiz_creation": PromptTemplates.get_quiz_creation_prompt,
|
109 |
-
"assignment_creation": PromptTemplates.get_assignment_creation_prompt,
|
110 |
-
"paper_creation": PromptTemplates.get_paper_creation_prompt,
|
111 |
-
"check_quiz": PromptTemplates.get_check_quiz_prompt,
|
112 |
-
"check_assignment": PromptTemplates.get_check_assignment_prompt,
|
113 |
-
"check_paper": PromptTemplates.get_check_paper_prompt
|
114 |
-
}
|
115 |
-
|
116 |
-
prompt_template = prompt_template_mapping.get(prompt_type, PromptTemplates.get_quiz_solving_prompt)()
|
117 |
-
|
118 |
-
trainer.set_custom_prompt_template(query_request.bot_id, prompt_template)
|
119 |
-
|
120 |
response, web_sources = trainer.get_response(
|
121 |
query_request.query, query_request.bot_id, query_request.chat_id
|
122 |
)
|
@@ -126,6 +108,7 @@ def send_query(query_request: QueryRequest):
|
|
126 |
raise HTTPException(status_code=500, detail=str(e))
|
127 |
|
128 |
|
|
|
129 |
@router.get("/list_chats/{bot_id}")
|
130 |
def list_chats(bot_id: str):
|
131 |
"""
|
|
|
99 |
Processes a query and returns the bot's response along with any web sources.
|
100 |
"""
|
101 |
try:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
102 |
response, web_sources = trainer.get_response(
|
103 |
query_request.query, query_request.bot_id, query_request.chat_id
|
104 |
)
|
|
|
108 |
raise HTTPException(status_code=500, detail=str(e))
|
109 |
|
110 |
|
111 |
+
|
112 |
@router.get("/list_chats/{bot_id}")
|
113 |
def list_chats(bot_id: str):
|
114 |
"""
|