merge with dev

Author: rohit-kadhe

.vscode/settings.json

@@ -1,7 +1,11 @@
-
 {
   "[python]": {
     "editor.defaultFormatter": "ms-python.black-formatter",
   },
   "editor.formatOnSave": true,
+  "python.testing.pytestArgs": [
+    "src"
+  ],
+  "python.testing.unittestEnabled": false,
+  "python.testing.pytestEnabled": true,
 }

README.md

@@ -255,6 +255,7 @@ Langtrace automatically captures traces from the following vendors:
 | Ollama       | Framework       | :x:                | :white_check_mark:              |
 | VertexAI     | Framework       | :x:                | :white_check_mark:              |
 | Vercel AI SDK| Framework       | :white_check_mark: | :x:                             |
+| EmbedChain   | Framework       | :x:                | :white_check_mark:              |
 | Pinecone     | Vector Database | :white_check_mark: | :white_check_mark:              |
 | ChromaDB     | Vector Database | :white_check_mark: | :white_check_mark:              |
 | QDrant       | Vector Database | :white_check_mark: | :white_check_mark:              |

mypy.ini

@@ -0,0 +1,16 @@
+[mypy]
+strict = True
+disable_error_code = import-untyped
+disallow_untyped_calls = True        # Disallow function calls without type annotations
+disallow_untyped_defs = True         # Disallow defining functions without type annotations
+disallow_any_explicit = True         # Disallow explicit use of `Any`
+disallow_any_generics = True         # Disallow generic types without specific type parameters
+disallow_incomplete_defs = True      # Disallow defining incomplete function signatures
+no_implicit_optional = True          # Disallow implicitly Optional types
+warn_unused_configs = True           # Warn about unused configurations
+warn_redundant_casts = True          # Warn about unnecessary type casts
+warn_return_any = True               # Warn if a function returns `Any`
+warn_unreachable = True              # Warn about unreachable code
+# Ignore external modules or allow specific imports
+follow_imports = skip
+ignore_missing_imports = True

pyproject.toml

@@ -12,10 +12,10 @@ 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.4',
@@ -28,40 +28,37 @@ 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==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",
+  "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",
+  "embedchain",
 ]
 
+test = ["pytest", "pytest-vcr", "pytest-asyncio"]
 
 
 [project.urls]
@@ -72,9 +69,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/anthropic_example/completion.py

@@ -7,7 +7,7 @@
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init()
+langtrace.init(write_spans_to_console=True)
 
 
 @with_langtrace_root_span("messages_create")

src/examples/crewai_example/instagram_post/__init__.py

@@ -0,0 +1,96 @@
+from crewai import Agent
+from tools.browser_tools import scrape_and_summarize_website
+from langchain_openai import ChatOpenAI
+# from langchain.llms import Ollama
+
+
+class MarketingAnalysisAgents:
+    def __init__(self):
+        # self.llm = Ollama(model=os.environ["MODEL"])
+        self.llm = ChatOpenAI(model_name="gpt-4", temperature=0.7)
+
+    def product_competitor_agent(self):
+        return Agent(
+            role="Lead Market Analyst",
+            goal="""\
+                Conduct amazing analysis of the products and
+                competitors, providing in-depth insights to guide
+                marketing strategies.""",
+            backstory="""\
+                As the Lead Market Analyst at a premier
+                digital marketing firm, you specialize in dissecting
+                online business landscapes.""",
+            tools=[scrape_and_summarize_website],
+            allow_delegation=False,
+            llm=self.llm,
+            verbose=True,
+        )
+
+    def strategy_planner_agent(self):
+        return Agent(
+            role="Chief Marketing Strategist",
+            goal="""\
+                Synthesize amazing insights from product analysis
+                to formulate incredible marketing strategies.""",
+            backstory="""\
+                You are the Chief Marketing Strategist at
+                a leading digital marketing agency, known for crafting
+                bespoke strategies that drive success.""",
+            tools=[scrape_and_summarize_website],
+            llm=self.llm,
+            verbose=True,
+        )
+
+    def creative_content_creator_agent(self):
+        return Agent(
+            role="Creative Content Creator",
+            goal="""\
+                Develop compelling and innovative content
+                for social media campaigns, with a focus on creating
+                high-impact Instagram ad copies.""",
+            backstory="""\
+                As a Creative Content Creator at a top-tier
+                digital marketing agency, you excel in crafting narratives
+                that resonate with audiences on social media.
+                Your expertise lies in turning marketing strategies
+                into engaging stories and visual content that capture
+                attention and inspire action.""",
+            tools=[scrape_and_summarize_website],
+            llm=self.llm,
+            verbose=True,
+        )
+
+    def senior_photographer_agent(self):
+        return Agent(
+            role="Senior Photographer",
+            goal="""\
+            Take the most amazing photographs for instagram ads that
+            capture emotions and convey a compelling message.""",
+            backstory="""\
+            As a Senior Photographer at a leading digital marketing
+            agency, you are an expert at taking amazing photographs that
+            inspire and engage, you're now working on a new campaign for a super
+            important customer and you need to take the most amazing photograph.""",
+            tools=[scrape_and_summarize_website],
+            llm=self.llm,
+            allow_delegation=False,
+            verbose=True,
+        )
+
+    def chief_creative_diretor_agent(self):
+        return Agent(
+            role="Chief Creative Director",
+            goal="""\
+                Oversee the work done by your team to make sure it's the best
+                possible and aligned with the product's goals, review, approve,
+                ask clarifying question or delegate follow up work if necessary to make
+                decisions""",
+            backstory="""\
+                You're the Chief Content Officer of leading digital
+                marketing specialized in product branding. You're working on a new
+                customer, trying to make sure your team is crafting the best possible
+                content for the customer.""",
+            tools=[scrape_and_summarize_website],
+            llm=self.llm,
+            verbose=True,
+        )

src/examples/crewai_example/instagram_post/agents.py

@@ -0,0 +1,80 @@
+
+from dotenv import load_dotenv
+
+load_dotenv()
+
+
+from langtrace_python_sdk import langtrace
+
+langtrace.init()
+
+from crewai import Crew
+from agents import MarketingAnalysisAgents
+from tasks import MarketingAnalysisTasks
+
+tasks = MarketingAnalysisTasks()
+agents = MarketingAnalysisAgents()
+
+print("## Welcome to the marketing Crew")
+print('-------------------------------')
+product_website = "https://langtrace.ai"
+product_details = "LLM App monitoring"
+
+
+# Create Agents
+product_competitor_agent = agents.product_competitor_agent()
+# strategy_planner_agent = agents.strategy_planner_agent()
+# creative_agent = agents.creative_content_creator_agent()
+# # Create Tasks
+website_analysis = tasks.product_analysis(product_competitor_agent, product_website, product_details)
+# market_analysis = tasks.competitor_analysis(product_competitor_agent, product_website, product_details)
+# campaign_development = tasks.campaign_development(strategy_planner_agent, product_website, product_details)
+# write_copy = tasks.instagram_ad_copy(creative_agent)
+
+# Create Crew responsible for Copy
+copy_crew = Crew(
+    agents=[
+        product_competitor_agent,
+        # strategy_planner_agent,
+        # creative_agent
+    ],
+    tasks=[
+        website_analysis,
+        # market_analysis,
+        # campaign_development,
+        # write_copy
+    ],
+    verbose=True
+)
+
+ad_copy = copy_crew.kickoff()
+
+# Create Crew responsible for Image
+# senior_photographer = agents.senior_photographer_agent()
+# chief_creative_diretor = agents.chief_creative_diretor_agent()
+# # Create Tasks for Image
+# take_photo = tasks.take_photograph_task(senior_photographer, ad_copy, product_website, product_details)
+# approve_photo = tasks.review_photo(chief_creative_diretor, product_website, product_details)
+
+# image_crew = Crew(
+#     agents=[
+#         senior_photographer,
+#         chief_creative_diretor
+#     ],
+#     tasks=[
+#         take_photo,
+#         approve_photo
+#     ],
+#     verbose=True
+# )
+
+# image = image_crew.kickoff()
+
+# Print results
+print("\n\n########################")
+print("## Here is the result")
+print("########################\n")
+print("Your post copy:")
+print(ad_copy)
+# print("'\n\nYour midjourney description:")
+# print(image)