fix file url

Chainlit · Dec 19, 2023 · ebfb37d · ebfb37d
1 parent b50da84
commit ebfb37d
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/backend/chainlit/data/__init__.py b/backend/chainlit/data/__init__.py
@@ -234,9 +234,12 @@ async def create_element(self, element: "Element"):
             "page": getattr(element, "page", None),
         }
 
+        if not element.for_id:
+            return
+
         await self.client.api.create_attachment(
             thread_id=element.thread_id,
-            step_id=element.for_id or "",
+            step_id=element.for_id,
             mime=element.mime,
             name=element.name,
             url=element.url,

diff --git a/libs/react-client/src/api/index.tsx b/libs/react-client/src/api/index.tsx
@@ -282,6 +282,9 @@ export class ChainlitAPI extends APIBase {
   getElementUrl(id: string, sessionId: string, accessToken?: string) {
     let queryParams = `?session_id=${sessionId}`;
     if (accessToken) {
+      if (accessToken.startsWith('Bearer ')) {
+        accessToken = accessToken.slice(7);
+      }
       queryParams += `&token=${accessToken}`;
     }
     return this.buildEndpoint(`/project/file/${id}${queryParams}`);