From f85dfa31e893a16cd3bc0ed80cd2b18503966b02 Mon Sep 17 00:00:00 2001
From: ogabrielluiz <gabriel@langflow.org>
Date: Wed, 5 Jun 2024 11:18:36 -0300
Subject: [PATCH] refactor: Add StreamURL and Log types to schema.py and update
 ChatOutputResponse in utils/schemas.py

---
 src/backend/base/langflow/schema/schema.py | 11 +++++++++++
 src/backend/base/langflow/utils/schemas.py |  1 +
 2 files changed, 12 insertions(+)

diff --git a/src/backend/base/langflow/schema/schema.py b/src/backend/base/langflow/schema/schema.py
index 921bd65b2..cc3886a92 100644
--- a/src/backend/base/langflow/schema/schema.py
+++ b/src/backend/base/langflow/schema/schema.py
@@ -5,6 +5,7 @@ from typing import Literal, Optional, cast
 from langchain_core.documents import Document
 from langchain_core.messages import AIMessage, BaseMessage, HumanMessage
 from pydantic import BaseModel, model_validator
+from typing_extensions import TypedDict
 
 
 class Record(BaseModel):
@@ -177,3 +178,13 @@ INPUT_FIELD_NAME = "input_value"
 
 InputType = Literal["chat", "text", "any"]
 OutputType = Literal["chat", "text", "any", "debug"]
+
+
+class StreamURL(TypedDict):
+    location: str
+
+
+class Log(TypedDict):
+    message: str | dict | StreamURL
+    type: str
+    type: str
diff --git a/src/backend/base/langflow/utils/schemas.py b/src/backend/base/langflow/utils/schemas.py
index cc7776460..647941f59 100644
--- a/src/backend/base/langflow/utils/schemas.py
+++ b/src/backend/base/langflow/utils/schemas.py
@@ -26,6 +26,7 @@ class ChatOutputResponse(BaseModel):
     stream_url: Optional[str] = None
     component_id: Optional[str] = None
     files: List[File] = []
+    type: str
 
     @field_validator("files", mode="before")
     def validate_files(cls, files):