From 0b783fb8e387963780216d15920e01758129f9a4 Mon Sep 17 00:00:00 2001
From: Abram <israelvictory87@gmail.com>
Date: Mon, 20 May 2024 19:57:11 +0100
Subject: [PATCH] Refactor (types.py): remove FuncResponse sdk type

---
 agenta-cli/agenta/sdk/types.py | 13 -------------
 1 file changed, 13 deletions(-)

diff --git a/agenta-cli/agenta/sdk/types.py b/agenta-cli/agenta/sdk/types.py
index 497cbdec39..d1b4ca8b39 100644
--- a/agenta-cli/agenta/sdk/types.py
+++ b/agenta-cli/agenta/sdk/types.py
@@ -10,19 +10,6 @@ def __init__(self, file_name: str, file_path: str):
         self.file_path = file_path
 
 
-class LLMTokenUsage(BaseModel):
-    completion_tokens: int
-    prompt_tokens: int
-    total_tokens: int
-
-
-class FuncResponse(BaseModel):
-    message: str
-    usage: Optional[LLMTokenUsage]
-    cost: Optional[float]
-    latency: float
-
-
 class DictInput(dict):
     def __new__(cls, default_keys=None):
         instance = super().__new__(cls, default_keys)