From 2c2c1cd49c0dc56bb20cd82d5bd13db5e77fcc02 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Helge-Mikael=20Nordg=C3=A5rd?=
 <helge-mikael.nordgard@arcticsoftware.no>
Date: Thu, 23 Jan 2025 20:04:51 +0100
Subject: [PATCH] Updated with try catch

---
 00_simple_chat_interface.py | 41 ++++++++++++++++++++++---------------
 1 file changed, 24 insertions(+), 17 deletions(-)

diff --git a/00_simple_chat_interface.py b/00_simple_chat_interface.py
index 6fc2d62..d1fc757 100644
--- a/00_simple_chat_interface.py
+++ b/00_simple_chat_interface.py
@@ -14,7 +14,7 @@ class Pipeline:
         OR_KEY: str
 
     def __init__(self):
-        self.name = "00 Basic Openrouter Chat"
+        self.name = "Basic Openrouter Chat"
         self.valves = self.Valves(
             **{
                 "pipelines": ["*"],
@@ -31,21 +31,28 @@ class Pipeline:
         pass
 
     def pipe(self, user_message: str, model_id: str, messages: List[dict], body: dict) -> Union[str, Generator, Iterator]:
-        response = requests.post(
-            url = self.valves.OR_URL,
-            headers = {
-                "Authorization": f"Bearer {self.valves.OR_KEY}"
-            },
+        try:
+            response = requests.post(
+                url = self.valves.OR_URL,
+                headers = {
+                    "Authorization": f"Bearer {self.valves.OR_KEY}"
+                },
 
-            data = json.dumps({
-                "model": self.valves.OR_MODEL,
-                "messages": [
-                    {
-                        "role": "user",
-                        "content": user_message
-                    }
-                ]
-            })
-        )
+                data = json.dumps({
+                    "model": self.valves.OR_MODEL,
+                    "messages": [
+                        {
+                            "role": "user",
+                            "content": user_message
+                        }
+                    ]
+                })
+            )
         
-        return response
\ No newline at end of file
+            return response
+        except requests.HTTPError as e:
+            logging.error(f"Clientresponse error: {e}")
+            return "HTTP backend error"
+        except Exception as e:
+            logging.error(f"Unexpected error: {e}")
+            return f"Unexpected error: {e}"
\ No newline at end of file