curation

Author: alizenhom

src/examples/crewai_example/simple_agent/main.py

@@ -7,8 +7,7 @@
 import agentops
 
 load_dotenv()
-agentops.init()
-langtrace.init(write_spans_to_console=False, batch=False)
+langtrace.init()
 
 
 class PoetryCrew:
@@ -32,15 +31,15 @@ def run(self):
 if __name__ == "__main__":
     print("## Welcome to Poetry Crew")
     print("-------------------------------")
-    topic = input(
-        dedent(
-            """
-      What topic do you want to write a poem on?
-    """
-        )
-    )
-
-    poetry_crew = PoetryCrew(topic=topic)
+    # topic = input(
+    #     dedent(
+    #         """
+    #   What topic do you want to write a poem on?
+    # """
+    #     )
+    # )
+
+    poetry_crew = PoetryCrew(topic="cold")
     result = poetry_crew.run()
     print("\n\n########################")
     print("## Here is you poem")

src/examples/crewai_example/simple_agent/tasks.py

@@ -18,4 +18,5 @@ def create_poem(self, agent, topic):
             ),
             expected_output="A creative and expressive poem that captures the essence of the given topic.",
             agent=agent,
+            output_file="poem.txt",
         )

src/examples/crewai_example/trip_planner/main.py

@@ -59,36 +59,36 @@ def run(self):
 if __name__ == "__main__":
     print("## Welcome to Trip Planner Crew")
     print("-------------------------------")
-    origin = input(
-        dedent(
-            """
-      From where will you be traveling from?
-    """
-        )
-    )
-    cities = input(
-        dedent(
-            """
-      What are the cities options you are interested in visiting?
-    """
-        )
-    )
-    date_range = input(
-        dedent(
-            """
-      What is the date range you are interested in traveling?
-    """
-        )
-    )
-    interests = input(
-        dedent(
-            """
-      What are some of your high level interests and hobbies?
-    """
-        )
-    )
+    # origin = input(
+    #     dedent(
+    #         """
+    #   From where will you be traveling from?
+    # """
+    #     )
+    # )
+    # cities = input(
+    #     dedent(
+    #         """
+    #   What are the cities options you are interested in visiting?
+    # """
+    #     )
+    # )
+    # date_range = input(
+    #     dedent(
+    #         """
+    #   What is the date range you are interested in traveling?
+    # """
+    #     )
+    # )
+    # interests = input(
+    #     dedent(
+    #         """
+    #   What are some of your high level interests and hobbies?
+    # """
+    #     )
+    # )
 
-    trip_crew = TripCrew(origin, cities, date_range, interests)
+    trip_crew = TripCrew("cairo", "marsa alam", "sep", "scuba diving")
     result = trip_crew.run()
     print("\n\n########################")
     print("## Here is you Trip Plan")

src/langtrace_python_sdk/instrumentation/crewai/instrumentation.py

@@ -20,6 +20,7 @@
 from typing import Collection
 from importlib_metadata import version as v
 from .patch import patch_crew
+import crewai
 
 
 class CrewAIInstrumentation(BaseInstrumentor):
@@ -46,7 +47,7 @@ def _instrument(self, **kwargs):
             )
             _W(
                 "crewai.task",
-                "Task.execute",
+                "Task.execute_sync",
                 patch_crew("Task.execute", version, tracer),
             )
         except Exception as e: