gabykim commited on
Commit
9644b48
·
1 Parent(s): 1760e74

disable polish question node for now

Browse files
src/know_lang_bot/chat_bot/chat_graph.py CHANGED
@@ -372,7 +372,8 @@ async def process_chat(
372
 
373
  try:
374
  result, _history = await chat_graph.run(
375
- PolishQuestionNode(),
 
376
  state=state,
377
  deps=deps
378
  )
@@ -398,7 +399,8 @@ async def stream_chat_progress(
398
  state = ChatGraphState(original_question=question)
399
  deps = ChatGraphDeps(collection=collection, config=config)
400
 
401
- start_node = PolishQuestionNode()
 
402
  history: list[HistoryStep[ChatGraphState, ChatResult]] = []
403
 
404
  try:
 
372
 
373
  try:
374
  result, _history = await chat_graph.run(
375
+ # Temporary fix to disable PolishQuestionNode
376
+ RetrieveContextNode(),
377
  state=state,
378
  deps=deps
379
  )
 
399
  state = ChatGraphState(original_question=question)
400
  deps = ChatGraphDeps(collection=collection, config=config)
401
 
402
+ # Temporary fix to disable PolishQuestionNode
403
+ start_node = RetrieveContextNode()
404
  history: list[HistoryStep[ChatGraphState, ChatResult]] = []
405
 
406
  try: