feat: get_output_text

Author: shreykx

genstack/genstack.py

@@ -29,6 +29,11 @@ async def __call(self, payload: Dict[str, Any]) -> Dict[str, Any]:
                 return {"error" : str(e)}
     async def __generate_async(self, payload : Dict[str, Any]) -> Dict[str, Any]:
         return await self.__call(payload=payload)
+    def __extract_first_text(self, result: dict) -> str:
+        for o in result.get("output", []):
+            if o.get("output", {}).get("type") == "TEXT":
+                return o["output"].get("text", "")
+        return ""
     def generate(self, input : Union[str, Dict[str, Any]], model : Optional[str] = "auto", track : Optional[str] = None) -> Dict[str, Any]:
         if not track :
             raise ValueError("Track is required.")
@@ -44,4 +49,8 @@ def generate(self, input : Union[str, Dict[str, Any]], model : Optional[str] = "
             "model": model,
             "track": track
         }
-        return asyncio.run(self.__generate_async(payload=payload))
+        return asyncio.run(self.__generate_async(payload=payload))
+    def get_output_text(self, input: Union[str, Dict[str, Any]], model: Optional[str] = "auto", track: Optional[str] = None) -> str:
+        result = self.generate(input=input, model=model, track=track)
+        first_text : str = self.__extract_first_text(result)
+        return first_text

pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "genstack"
-version = "0.1.9"
+version = "0.2.0"
 description = "Universal AI SDK from Genstack"
 authors = [{name="Shrey Kumar", email="shreyk.dev@gmail.com"}]
 readme = "README.md"

tests/call_test.py

@@ -9,9 +9,6 @@
 client = Genstack(api_key=os.getenv("GENSTACK_API_KEY"))
 
 
-res = client.generate(input="3 fun facts about Ferrari", track="dragon-track", model="gpt-4-1-nano-oai")
+res = client.get_output_text(input="3 fun facts about Ferrari", track="ultimate-track", model="gpt-4o-oai")
 
-if "output" in res:
-    print(res["output"][0]["output"]["text"])
-else:
-    print("Error:", res.get("message", "Unknown error"))
+print(res)