shoom013 commited on
Commit
76acc8f
·
verified ·
1 Parent(s): 0943fdc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -126,14 +126,14 @@ def rag(input_text, history, jezik):
126
  query_engine = index2.as_query_engine()
127
  # return query_engine.query(input_text)
128
  # return history.append({"role": "assistant", "content": query_engine.query(input_text)})
129
- return history.append(
130
  ChatMessage(role="user",
131
  content=input_text
132
  ).append(
133
  role="assistant",
134
  content=query_engine.query(input_text)
135
  )
136
- )
137
  # collection.add(
138
  # documents=documents,
139
  # ids=[f"id{last+i}" for i in range(len(documents))],
@@ -160,11 +160,11 @@ def rag(input_text, history, jezik):
160
  # if (o_jezik!='N/A'):
161
  # input_text += " - odgovori " + o_jezik + "."
162
  # return query_engine.query(input_text)
163
- return history.append(
164
  ChatMessage(role="assistant",
165
  content=chat_engine.chat(input_text)
166
  )
167
- )
168
  # Interface
169
  # gr.Textbox(label="Pitanje:", lines=6),
170
  # outputs=[gr.Textbox(label="Odgovor:", lines=6)],
 
126
  query_engine = index2.as_query_engine()
127
  # return query_engine.query(input_text)
128
  # return history.append({"role": "assistant", "content": query_engine.query(input_text)})
129
+ return [history.append(
130
  ChatMessage(role="user",
131
  content=input_text
132
  ).append(
133
  role="assistant",
134
  content=query_engine.query(input_text)
135
  )
136
+ )]
137
  # collection.add(
138
  # documents=documents,
139
  # ids=[f"id{last+i}" for i in range(len(documents))],
 
160
  # if (o_jezik!='N/A'):
161
  # input_text += " - odgovori " + o_jezik + "."
162
  # return query_engine.query(input_text)
163
+ return [history.append(
164
  ChatMessage(role="assistant",
165
  content=chat_engine.chat(input_text)
166
  )
167
+ )]
168
  # Interface
169
  # gr.Textbox(label="Pitanje:", lines=6),
170
  # outputs=[gr.Textbox(label="Odgovor:", lines=6)],