Srushti-Kamble commited on
Commit
795b664
·
1 Parent(s): 5e9fce0

chat.py works agent imported

Browse files
Files changed (1) hide show
  1. backend/app/routes/chat.py +4 -1
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,