Spaces:
Running
Running
Srushti-Kamble commited on
Commit ·
795b664
1
Parent(s): 5e9fce0
chat.py works agent imported
Browse files
backend/app/routes/chat.py
CHANGED
|
@@ -22,7 +22,6 @@ from app.models import User, ChatMessage, Document
|
|
| 22 |
from app.metrics import record_query_response_time
|
| 23 |
from app.schemas import ChatRequest, ChatResponse, ChatMessageResponse, ChatHistoryResponse, SourceChunk
|
| 24 |
from app.auth import get_current_user
|
| 25 |
-
from app.rag.agent import generate_answer, generate_answer_stream
|
| 26 |
from app.rate_limit import limiter
|
| 27 |
|
| 28 |
logger = logging.getLogger(__name__)
|
|
@@ -84,6 +83,8 @@ def ask_question(
|
|
| 84 |
)
|
| 85 |
|
| 86 |
# Generate answer
|
|
|
|
|
|
|
| 87 |
result = generate_answer(
|
| 88 |
question=payload.question,
|
| 89 |
user_id=user.id,
|
|
@@ -173,6 +174,8 @@ def ask_question_stream(
|
|
| 173 |
sources = []
|
| 174 |
|
| 175 |
try:
|
|
|
|
|
|
|
| 176 |
for chunk in generate_answer_stream(
|
| 177 |
question=payload.question,
|
| 178 |
user_id=user.id,
|
|
|
|
| 22 |
from app.metrics import record_query_response_time
|
| 23 |
from app.schemas import ChatRequest, ChatResponse, ChatMessageResponse, ChatHistoryResponse, SourceChunk
|
| 24 |
from app.auth import get_current_user
|
|
|
|
| 25 |
from app.rate_limit import limiter
|
| 26 |
|
| 27 |
logger = logging.getLogger(__name__)
|
|
|
|
| 83 |
)
|
| 84 |
|
| 85 |
# Generate answer
|
| 86 |
+
from app.rag.agent import generate_answer
|
| 87 |
+
|
| 88 |
result = generate_answer(
|
| 89 |
question=payload.question,
|
| 90 |
user_id=user.id,
|
|
|
|
| 174 |
sources = []
|
| 175 |
|
| 176 |
try:
|
| 177 |
+
from app.rag.agent import generate_answer_stream
|
| 178 |
+
|
| 179 |
for chunk in generate_answer_stream(
|
| 180 |
question=payload.question,
|
| 181 |
user_id=user.id,
|