Merge branch 'main' of github.com:Scale3-Labs/langtrace-python-sdk into release

Author: karthikscale3

README.md

@@ -246,6 +246,7 @@ Langtrace automatically captures traces from the following vendors:
 | Groq         | LLM             | :x:                | :white_check_mark:              |
 | Perplexity   | LLM             | :white_check_mark: | :white_check_mark:              |
 | Gemini       | LLM             | :x:                | :white_check_mark:              |
+| Mistral      | LLM             | :x:                | :white_check_mark:              |
 | Langchain    | Framework       | :x:                | :white_check_mark:              |
 | LlamaIndex   | Framework       | :white_check_mark: | :white_check_mark:              |
 | Langgraph    | Framework       | :x:                | :white_check_mark:              |

pyproject.toml

@@ -12,13 +12,13 @@ description = "Python SDK for LangTrace"
 readme = "README.md"
 authors = [{ name = "Scale3 Labs", email = "[email protected]" }]
 license = "Apache-2.0"
-classifiers=[
-    "Programming Language :: Python :: 3",
-    "License :: OSI Approved :: Apache Software License",
-    "Operating System :: OS Independent",
+classifiers = [
+  "Programming Language :: Python :: 3",
+  "License :: OSI Approved :: Apache Software License",
+  "Operating System :: OS Independent",
 ]
 dependencies = [
-  'trace-attributes==7.0.1',
+  'trace-attributes==7.0.4',
   'opentelemetry-api>=1.25.0',
   'opentelemetry-sdk>=1.25.0',
   'opentelemetry-instrumentation>=0.47b0',
@@ -28,39 +28,36 @@ dependencies = [
   'tiktoken>=0.1.1',
   'colorama>=0.4.6',
   'sqlalchemy',
-  'fsspec>=2024.6.0'
+  'fsspec>=2024.6.0',
+  "transformers>=4.11.3",
 ]
 
 requires-python = ">=3.9"
 
 [project.optional-dependencies]
 dev = [
-    "openai",
-    "anthropic",
-    "chromadb",
-    "qdrant-client",
-    "python-dotenv",
-    "pinecone-client",
-    "langchain",
-    "langchain-community",
-    "langchain-openai",
-    "langchain-openai",
-    "chromadb",
-    "cohere",
-    "qdrant_client",
-    "weaviate-client",
-    "ollama",
-    "groq",
-    "google-generativeai",
-    "google-cloud-aiplatform"
-]
-
-test = [
-    "pytest",
-    "pytest-vcr",
-    "pytest-asyncio",
+  "openai==1.30.1",
+  "anthropic",
+  "chromadb",
+  "qdrant-client",
+  "python-dotenv",
+  "pinecone-client",
+  "langchain",
+  "langchain-community",
+  "langchain-openai",
+  "langchain-openai",
+  "chromadb",
+  "cohere",
+  "qdrant_client",
+  "weaviate-client",
+  "ollama",
+  "groq",
+  "google-generativeai",
+  "google-cloud-aiplatform",
+  "mistralai",
 ]
 
+test = ["pytest", "pytest-vcr", "pytest-asyncio"]
 
 
 [project.urls]
@@ -71,9 +68,7 @@ Homepage = "https://github.com/Scale3-Labs/langtrace-python-sdk"
 path = "src/langtrace_python_sdk/version.py"
 
 [tool.hatch.build.targets.sdist]
-include = [
-  "/src",
-]
+include = ["/src"]
 
 [tool.hatch.build.targets.wheel]
 packages = ["src/langtrace_python_sdk", "src/examples", "src/tests"]

src/examples/crewai_example/__init__.py

@@ -0,0 +1,32 @@
+from crewai import Agent
+from langchain_openai import ChatOpenAI
+from langchain_anthropic import ChatAnthropic
+from langchain_cohere import ChatCohere
+from langchain_ollama import ChatOllama
+
+
+class PoetryAgents:
+    def __init__(self):
+        self.open_ai = ChatOpenAI(
+            model_name="gpt-4", temperature=0.7, stream_usage=True
+        )
+        self.anthropic = ChatAnthropic(
+            model_name="claude-3-5-sonnet-20240620", temperature=0.7
+        )
+
+        self.cohere = ChatCohere(model="command-r", temperature=0.7)
+        self.ollama = ChatOllama(model="llama3", temperature=0.7)
+
+    def create_poet_agent(self):
+        return Agent(
+            role="Expert Poetry Writer",
+            backstory="""
+                    I am an Expert in poetry writing and creative expression.
+                    I have been writing poetry for over 10 years and have published several collections.
+                    I have a deep understanding of various poetic forms, styles, and themes. I am here to help you create beautiful and meaningful poetry that resonates with your emotions and experiences.
+                """,
+            goal="""Create a poem that captures the essence of a given theme or emotion""",
+            allow_delegation=False,
+            verbose=True,
+            llm=self.open_ai,
+        )

src/examples/crewai_example/simple_agent/__init__.py

@@ -0,0 +1,38 @@
+from crewai import Crew
+from .agents import PoetryAgents
+from .tasks import PoetryTasks
+from langtrace_python_sdk import langtrace
+from dotenv import load_dotenv
+
+load_dotenv()
+langtrace.init()
+
+
+class PoetryCrew:
+    def __init__(self, topic) -> None:
+        self.topic = topic
+
+    def run(self):
+        agents = PoetryAgents()
+        tasks = PoetryTasks()
+
+        poetry_agent = agents.create_poet_agent()
+
+        create_poem = tasks.create_poem(poetry_agent, self.topic)
+
+        crew = Crew(agents=[poetry_agent], tasks=[create_poem], verbose=True)
+        res = crew.kickoff()
+        return res
+
+
+# This is the main function that you will use to run your custom crew.
+# You can run this file using `python -m src.examples.crewai_example.simple_agent.main`
+if __name__ == "__main__":
+    print("## Welcome to Poetry Crew")
+    print("-------------------------------")
+    poetry_crew = PoetryCrew(topic="cold")
+    result = poetry_crew.run()
+    print("\n\n########################")
+    print("## Here is you poem")
+    print("########################\n")
+    print(result)

src/examples/crewai_example/simple_agent/agents.py

@@ -0,0 +1,22 @@
+from crewai import Task
+from textwrap import dedent
+
+
+class PoetryTasks:
+    def create_poem(self, agent, topic):
+        return Task(
+            description=dedent(
+                f"""
+            **Task**: Create a Poem on {topic}
+            **Description**: Write a poem on the given topic that captures the essence of the theme or emotion. 
+                The poem should be creative, expressive, and resonate with the reader's emotions and experiences. 
+                Your poem should be well-structured, engaging, and evoke a sense of beauty and meaning.
+
+            **Parameters**: 
+            - Topic: {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/simple_agent/main.py

@@ -0,0 +1,54 @@
+from crewai import Agent
+from langchain_openai import ChatOpenAI
+from langchain_ollama import ChatOllama
+from langchain_cohere import ChatCohere
+from langchain_anthropic import ChatAnthropic
+from dotenv import load_dotenv
+
+load_dotenv()
+
+
+class TravelAgents:
+    def __init__(self):
+        self.OpenAIGPT35 = ChatOpenAI(model_name="gpt-3.5-turbo", temperature=0.7)
+        self.OpenAIGPT4 = ChatOpenAI(model_name="gpt-4", temperature=0.7)
+        self.Ollama = ChatOllama(model="llama3")
+        self.Cohere = ChatCohere(model="command-r")
+        self.Anthropic = ChatAnthropic(model="claude-3-5-sonnet-20240620")
+
+    def expert_travel_agent(self):
+        return Agent(
+            role="Expert Travel Agent",
+            backstory="""
+                    I am an Expert in travel planning and itinerary creation.
+                    I have been in the travel industry for over 10 years and have helped thousands of clients plan their dream vacations.
+                    I have extensive knowledge of popular travel destinations, local attractions, and travel logistics. I am here to help you create a personalized travel itinerary that suits your preferences and budget.
+                """,
+            goal="""Create a 7 day travel itinerary with detailed per-day plans, include budget, packing suggestions, and local/safety tips.""",
+            # tools=[tool_1, tool_2],
+            allow_delegation=False,
+            verbose=True,
+            llm=self.Cohere,
+        )
+
+    def city_selection_expert(self):
+        return Agent(
+            role="City Selection Expert",
+            backstory="""Expert at analyzing and selecting the best cities for travel based on data""",
+            goal="""Select the best cities based on weather, season, prices and traveler preferences""",
+            # tools=[tool_1, tool_2],
+            allow_delegation=False,
+            verbose=True,
+            llm=self.Cohere,
+        )
+
+    def local_tour_guide(self):
+        return Agent(
+            role="Local Expert at this city",
+            goal="Provide the BEST insights about the selected city",
+            backstory="""A knowledgeable local guide with extensive information about the city, it's attractions and customs""",
+            # tools=[tool_1, tool_2],
+            allow_delegation=False,
+            verbose=True,
+            llm=self.Cohere,
+        )

src/examples/crewai_example/simple_agent/tasks.py

@@ -0,0 +1,67 @@
+from crewai import Crew
+from .agents import TravelAgents
+from .tasks import TravelTasks
+from langtrace_python_sdk import langtrace
+from dotenv import load_dotenv
+
+load_dotenv()
+
+langtrace.init()
+
+
+class TripCrew:
+    def __init__(self, origin, cities, date_range, interests):
+        self.origin = origin
+        self.cities = cities
+        self.date_range = date_range
+        self.interests = interests
+
+    def run(self):
+        # Define your custom agents and tasks in agents.py and tasks.py
+        agents = TravelAgents()
+        tasks = TravelTasks()
+
+        # Define your custom agents and tasks here
+        expert_travel_agent = agents.expert_travel_agent()
+        city_selection_expert = agents.city_selection_expert()
+        local_tour_guide = agents.local_tour_guide()
+
+        # Custom tasks include agent name and variables as input
+        plan_itinerary = tasks.plan_itinerary(
+            expert_travel_agent, self.cities, self.date_range, self.interests
+        )
+
+        identify_city = tasks.identify_city(
+            city_selection_expert,
+            self.origin,
+            self.cities,
+            self.interests,
+            self.date_range,
+        )
+
+        gather_city_info = tasks.gather_city_info(
+            local_tour_guide, self.cities, self.date_range, self.interests
+        )
+
+        # Define your custom crew here
+        crew = Crew(
+            agents=[expert_travel_agent, city_selection_expert, local_tour_guide],
+            tasks=[plan_itinerary, identify_city, gather_city_info],
+            verbose=True,
+        )
+
+        result = crew.kickoff()
+        return result
+
+
+# This is the main function that you will use to run your custom crew.
+if __name__ == "__main__":
+    print("## Welcome to Trip Planner Crew")
+    print("-------------------------------")
+
+    trip_crew = TripCrew("cairo", "marsa alam", "sep", "scuba diving")
+    result = trip_crew.run()
+    print("\n\n########################")
+    print("## Here is you Trip Plan")
+    print("########################\n")
+    print(result)