Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)],
|