mtyrrell commited on
Commit
ca43139
·
1 Parent(s): db22ea7
Files changed (1) hide show
  1. app/main.py +7 -6
app/main.py CHANGED
@@ -397,12 +397,13 @@ add_routes(
397
  output_type=ChatFedOutput
398
  )
399
 
400
- # add_routes(
401
- # app,
402
- # RunnableLambda(flexible_handler),
403
- # path="/chatfed",
404
- # config={"enable_streaming": True}
405
- # )
 
406
 
407
  # ChatUI-compatible streaming route (yields tokens via SSE)
408
  # add_routes(
 
397
  output_type=ChatFedOutput
398
  )
399
 
400
+ add_routes(
401
+ app,
402
+ RunnableLambda(lambda d: flexible_handler(d).result), # returns just the string
403
+ path="/chatfed",
404
+ config={"enable_streaming": True, "exclude_run_metadata": True},
405
+ output_type=str
406
+ )
407
 
408
  # ChatUI-compatible streaming route (yields tokens via SSE)
409
  # add_routes(