Spaces:
Running
Running
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)],
|