improve model

Tauffer-Consulting · Jun 21, 2024 · 4a2b33d · 4a2b33d
1 parent 53deb5a
commit 4a2b33d
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 deletions.
diff --git a/pieces/GetGoogleSheetPiece/models.py b/pieces/GetGoogleSheetPiece/models.py
@@ -13,12 +13,13 @@ class InputModel(BaseModel):
 
 
 class OutputModel(BaseModel):
-    output_data: dict = Field(
-        description="Sleep piece executed"
+    output_data: list = Field(
+        title="Output Data.",
+        description="List of records from the Google Sheet."
     )
 
 class SecretsModel(BaseModel):
     service_account_data: str = Field(
         title="Service Account",
-        description="Service account JSON data for Google Sheets API",
+        description="Service account JSON data for Google Sheets API. Paste it as a string.",
     )
diff --git a/pieces/GetGoogleSheetPiece/piece.py b/pieces/GetGoogleSheetPiece/piece.py
@@ -25,5 +25,8 @@ def piece_function(self, input_data: InputModel, secrets_data: SecretsModel):
         records = worksheet.get_all_records()
         df = pd.DataFrame.from_records(records)
 
-        return OutputModel(output_data=df.to_json())
+        print('DF', df.to_dict())
+        print('df', df.to_dict(orient='records'))
+
+        return OutputModel(output_data=df.to_dict(orient='records'))