[FEAT][Output file types]

example.py

@@ -38,7 +38,7 @@ agent = Agent(
     auto_generate_prompt=False,  # Auto generate prompt for the agent based on name, description, and system prompt, task
     artifacts_on=True,
     artifacts_output_path="roth_ira_report",
-    artifacts_file_extension=".md",
+    artifacts_file_extension=".txt",
     max_tokens=8000,
 )
 

pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "swarms"
-version = "6.0.3"
+version = "6.0.4"
 description = "Swarms - Pytorch"
 license = "MIT"
 authors = ["Kye Gomez <kye@apac.ai>"]

roth_ira_report.pdf

@@ -1,87 +0,0 @@
-%PDF-1.3
-%“Œ‹ž ReportLab Generated PDF document http://www.reportlab.com
-1 0 obj
-<<
-/F1 2 0 R
->>
-endobj
-2 0 obj
-<<
-/BaseFont /Helvetica /Encoding /WinAnsiEncoding /Name /F1 /Subtype /Type1 /Type /Font
->>
-endobj
-3 0 obj
-<<
-/Contents 8 0 R /MediaBox [ 0 0 612 792 ] /Parent 7 0 R /Resources <<
-/Font 1 0 R /ProcSet [ /PDF /Text /ImageB /ImageC /ImageI ]
->> /Rotate 0 /Trans <<
-
->> 
-  /Type /Page
->>
-endobj
-4 0 obj
-<<
-/Contents 9 0 R /MediaBox [ 0 0 612 792 ] /Parent 7 0 R /Resources <<
-/Font 1 0 R /ProcSet [ /PDF /Text /ImageB /ImageC /ImageI ]
->> /Rotate 0 /Trans <<
-
->> 
-  /Type /Page
->>
-endobj
-5 0 obj
-<<
-/PageMode /UseNone /Pages 7 0 R /Type /Catalog
->>
-endobj
-6 0 obj
-<<
-/Author (anonymous) /CreationDate (D:20241104180641-05'00') /Creator (ReportLab PDF Library - www.reportlab.com) /Keywords () /ModDate (D:20241104180641-05'00') /Producer (ReportLab PDF Library - www.reportlab.com) 
-  /Subject (unspecified) /Title (untitled) /Trapped /False
->>
-endobj
-7 0 obj
-<<
-/Count 2 /Kids [ 3 0 R 4 0 R ] /Type /Pages
->>
-endobj
-8 0 obj
-<<
-/Filter [ /ASCII85Decode /FlateDecode ] /Length 2270
->>
-stream
-GasaqgMYb8&:O"K%$3S2L?sf/fQ#c4M[Q:SMomN,!e+:76YU%nX6qU:W*eF<92_Q+P6m1-1,#tT3HO>A+r]Jh$GYVkkl$Bf^5XP0cmkl*Fuc>JDtIc:'<XjuU[p3Gj52>aZWF>:d?8^nqKMSRH*tr)gZi68MO)8#l#+&)e6'<V-@Vc&@aE?_>^/c=88+j6dX*T+e_\!5eJDD[^Mq\E/k<U&DfTmY's)atOtuab'`HV&_:mS:p$^l+Y"81k-`+rsN7XB779[X7.1?1FL,@\^S_99rZ=M];SD#2g7sj+X3+MM`j[j$O3B5G7RuY.tT+X3aUV3tePd'?kHT/g<3%GZaR?JoTb$h>ZN/K@O.$qXDHBDBCg/rEHc]Yif[=e2-`2\G0"66M)QaGJ`D]=R$VY$9/(p7ZmkO:)``D+<4jVihC@]MXrX[;P;di6!ORJ+ZuFZQ+Eq^0n:F)ZlbFVuiZ1OjO'm]GO"ot7D:F@grs$!TAjKY0.Eop`Z65;-b"=f56!.cup-*DItcL^em0f;1u6.AFP<GYf-=ZQt[tOcPa4ntfo!Q]'bo4#<pH\Y/P8!;thWb4J=Th>-lY-TrRU4dC%1F49Zg(B87[04t4ecuc9bNF##7ZMYrF7BQU!)1U1PJFQ!j,P5"rn&Y`j<<35mN\3kc>TClO]XeIJd>eU"q;]94j*l[*hf]Pd\YN"Q-L!l,em?8clRIId50+*Ene1nh1OG#4HA0j+:!d85q7N>hk-ML2G6(G1S'KD*P.lE>Bdku4<AKhe_;Y.h5WDhT&%qO[n]Ms:-[:C<Xr4H,H0;Yej47at)t7cDAW>o*DUHRinRPf^Gc-"TM-hsOd7)uIh,ZNOc3Wh+d"?j6$<"/^<"leUf?!=3dd(TCFu2\QKGikH>ZiRSU#C/43X.60(On0NnCb'qBbH,dqS=kH`jP-GO3DKFUiu:\a`IKoDoY#/ajjI`Qhf)=JPKNYNTQ%;#Od_43Uh!Bf3Vg>M`M:YOp4i((AkUsd]e.Edi[T9WZOC1H&J6t9J6.uZL24Z$abSni)O4&M,a^*qd$Zc9CSWDpK$Z9=-P:5SXSL5SJrEXZgQJ:?=LLt"6Iu>h:7!A4c7)MW8H/C?r#+O/l[D<P(>'Qf00<]+l@@\0kc9lWd$p>ei^6,lm'VpH?r5nDcL=ddSt14RLJj7E0nFM:8::CWn87Qj2r41"0p'Y0K2["Alkm/0HMI(80&V$m9[g83LgS<]P0P"-6[]!UK'D]!%rH,9ZO/iqc7?4kf#5ugNuicHpg&3+>]h0J4sU5O92s-*pUn+UN"Ml$!Fk&\?s,D7@C@4NZ-27(u1;)JQ=QIceDs]k!6R\"7arPE@UM8Q7sC?YSLA!-\apMb0Al2WFk'L;f9&B6oUYpm>QMnLdhDCSob%1f&W)r_,#]8JQSXlCN4'])"n$+?j:MRI6h-uO92AR/e[=/;"m`jVVS&>Fu)]j%K)6*ODOkm`*C:18/ndaUlLWN`Dnis@S]C'i1o5<mcY0)L4aTKE.2M2Tm_mN_NhPFZgn6@/0`^1YrYYPA83l:jK2Z%69eM_a8SUWf)q.9'D-)nOa]3B7O&-,/](2G8=-t$iJl5nH*@$?\Y5Y_'"N>LHB?\o6&s0GkYlFGj?c_gY4-V4FrLO[1uS:J_"gPlPb;(5I0R@U,:#7*?I3kt>$O<\$"8TH"$Q[$kbqHh]]_Bg&rn^lgK+s_P-A2e6;S'fIEY2?01a]-6F69]KG_J""*9asTEOVDX0?dT[VP08m0%iDNS'_Uan<<?5>Jcj@a7!Hb6<-["&Fe1)(UZ]'OVFBH_BHIgU8HDRqD:>oN;tr)19r:'D',po(,BGn4&T4=-744cs)cK\LUTp\H&0Y3,k+L:bQ#jD0?IDkY7Z$Xn*>U>Zb*'*8[e^h@3\BPb%2,9ESHCRWlb3L*?hV)qq!\8fF5m/c9&O67>qG2`dgg3^2AMCJ&o]juRT4+G%*/1o$E3pk\DoWS\eM1(W+nlJ,N5gJ,)=T1u'@Z$PY/WeIo[q#W)5h,Ng6N>Qp]nR,F)QTAmejpUK=6W(cC5g"W<7>CT-oKf<"j?o===)4ejC(l8r*RGW1XttHO<mN+.+^t&WdoR3b*=%-oB`R]"3s6<nq?\\kG6i$,O9"bo$)[U7OLZ6Q<OW,.7(+\I+G4c%KT.J/s+Nm3"B4t6!d:h`d)7,W[$#\M13U.>8(NCcK-1&S?+4WmmfB`+TM[\2UOQk38WeWo$oW2t=GI&_R6";'j4bm$<C737!;rt\_Y:gukU)[GqB+6MmF_~>endstream
-endobj
-9 0 obj
-<<
-/Filter [ /ASCII85Decode /FlateDecode ] /Length 272
->>
-stream
-Gap@Hd7T$G%#+0E'Z[g7ZE5nP>!ZTTe`[B_?#$_5Dedb9e$h8nKEMq%Et33B=Ll0a-fDsRrT<#Zi"Omh[=s+NX'aN8rM,Dm4B_Of9:oA7W(QBS9a2\2&J/NRNA;716'+BL0O5h.Ucf\MXI77eJU?W!]DXOK1h`5##+&cF7&?AO3ML.Q:mK+qPHEdO#:btJlO(7WDjc`?`MC$^g[I4mSu!jZZ]JlpcB_ZYiQ;q7Pp.T9Q:P$EObbV\N<ieu52Y/4TA]0es2O%'FGF1%~>endstream
-endobj
-xref
-0 10
-0000000000 65535 f 
-0000000073 00000 n 
-0000000104 00000 n 
-0000000211 00000 n 
-0000000404 00000 n 
-0000000597 00000 n 
-0000000665 00000 n 
-0000000961 00000 n 
-0000001026 00000 n 
-0000003387 00000 n 
-trailer
-<<
-/ID 
-[<9407f775befdeb4ef8b5cf72ca950654><9407f775befdeb4ef8b5cf72ca950654>]
-% ReportLab generated PDF document -- digest (http://www.reportlab.com)
-
-/Info 6 0 R
-/Root 5 0 R
-/Size 10
->>
-startxref
-3749
-%%EOF

swarms/artifacts/main_artifact.py

@@ -33,6 +33,7 @@ class Artifact(BaseModel):
     Represents a file artifact.
 
     Attributes:
+        folder_path
         file_path (str): The path to the file.
         file_type (str): The type of the file.
         contents (str): The contents of the file.
@@ -272,28 +273,27 @@ class Artifact(BaseModel):
         if output_format == ".pdf":
             self._save_as_pdf(output_path)
         else:
-            with open(output_path, "w", encoding="utf-8"):
+            if output_format == ".md":
-                if output_format == ".md":
+                # Create the file in the specified folder
-                    # Create the file in the specified folder
+                create_file_in_folder(
-                    create_file_in_folder(
+                    self.folder_path,
-                        self.folder_path,
+                    self.file_path,
-                        self.file_path,
+                    f"{os.path.basename(self.file_path)}\n\n{self.contents}",
-                        f"{os.path.basename(self.file_path)}\n\n{self.contents}",
+                )
-                    )
 
-                elif output_format == ".py":
+            elif output_format == ".py":
-                    # Add Python file header
+                # Add Python file header
-                    create_file_in_folder(
+                create_file_in_folder(
-                        self.folder_path,
+                    self.folder_path,
-                        self.file_path,
+                    self.file_path,
-                        f"#{os.path.basename(self.file_path)}\n\n{self.contents}",
+                    f"#{os.path.basename(self.file_path)}\n\n{self.contents}",
-                    )
+                )
-                else:  # .txt
+            else:  # .txt
-                    create_file_in_folder(
+                create_file_in_folder(
-                        self.folder_path,
+                    self.folder_path,
-                        self.file_path,
+                    self.file_path,
-                        self.contents,
+                    self.contents,
-                    )
+                )
 
     def _save_as_pdf(self, output_path: str) -> None:
         """