From e3f6281bf9f6642ce76c0691b342cd30d9e49cee Mon Sep 17 00:00:00 2001
From: Ramiro Medina <64783088+ramedina86@users.noreply.github.com>
Date: Fri, 16 Aug 2024 15:21:38 +0000
Subject: [PATCH] fix: Types

---
 src/writer/ai.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/writer/ai.py b/src/writer/ai.py
index 810b195ad..d965bef6d 100644
--- a/src/writer/ai.py
+++ b/src/writer/ai.py
@@ -29,11 +29,11 @@
     GraphRemoveFileFromGraphResponse,
     GraphUpdateResponse,
     StreamingData,
-    Input
 )
 from writerai.types import File as SDKFile
 from writerai.types import Graph as SDKGraph
 from writerai.types.chat_chat_params import Message as WriterAIMessage
+from writerai.types.application_generate_content_params import Input
 
 from writer.core import get_app_process