Prathamesh1420 commited on
Commit
36b8083
·
verified ·
1 Parent(s): cdf7e24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -27,10 +27,14 @@ else:
27
  # User input with text_area for multi-line input
28
  user_input = st.text_area(
29
  "Enter your query:",
30
- value="Get me a list of RAG papers from 2024. Include the source, title, author, publication date, a brief summary, githublink and a link to each paper.",
31
  height=150
32
  )
33
 
 
 
 
 
34
  # Initialize the arxiv_qa_assistant
35
  assistants = taskingai.assistant.list_assistants()
36
  arxiv_qa_assistant = next((assistant for assistant in assistants if assistant.name == "arxivagent"), None)
@@ -43,7 +47,7 @@ else:
43
  user_message = taskingai.assistant.create_message(
44
  assistant_id=arxiv_qa_assistant.assistant_id,
45
  chat_id=new_chat.chat_id,
46
- text=user_input
47
  )
48
 
49
  assistant_message = taskingai.assistant.generate_message(
 
27
  # User input with text_area for multi-line input
28
  user_input = st.text_area(
29
  "Enter your query:",
30
+ value="Get me a list of RAG papers from 2024.",
31
  height=150
32
  )
33
 
34
+ # Append fixed information to ensure these details are always included
35
+ fixed_details = " Include the source, title, author, publication date, a brief summary, GitHub link, and a link to each paper."
36
+ full_query = user_input.strip() + fixed_details
37
+
38
  # Initialize the arxiv_qa_assistant
39
  assistants = taskingai.assistant.list_assistants()
40
  arxiv_qa_assistant = next((assistant for assistant in assistants if assistant.name == "arxivagent"), None)
 
47
  user_message = taskingai.assistant.create_message(
48
  assistant_id=arxiv_qa_assistant.assistant_id,
49
  chat_id=new_chat.chat_id,
50
+ text=full_query
51
  )
52
 
53
  assistant_message = taskingai.assistant.generate_message(