langtrace-python-sdk 2.1.13__tar.gz → 2.1.14__tar.gz

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: langtrace-python-sdk
-Version: 2.1.13
+Version: 2.1.14
 Summary: Python SDK for LangTrace
 Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
 Author-email: Scale3 Labs <engineering@scale3labs.com>
@@ -19,7 +19,7 @@ Requires-Dist: opentelemetry-instrumentation>=0.46b0
 Requires-Dist: opentelemetry-sdk>=1.25.0
 Requires-Dist: sqlalchemy
 Requires-Dist: tiktoken>=0.1.1
-Requires-Dist: trace-attributes==4.0.5
+Requires-Dist: trace-attributes<6.0.0,>=5.0.0
 Provides-Extra: dev
 Requires-Dist: anthropic; extra == 'dev'
 Requires-Dist: chromadb; extra == 'dev'
@@ -193,7 +193,7 @@ langtrace.init(custom_remote_exporter=<your_exporter>, batch=<True or False>)
 - `@with_langtrace_root_span` - this decorator is designed to organize and relate different spans, in a hierarchical manner. When you're performing multiple operations that you want to monitor together as a unit, this function helps by establishing a "parent" (`LangtraceRootSpan` or whatever is passed to `name`) span. Then, any calls to the LLM APIs made within the given function (fn) will be considered "children" of this parent span. This setup is especially useful for tracking the performance or behavior of a group of operations collectively, rather than individually.
 
 ```python
-from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
+from langtrace_python_sdk import with_langtrace_root_span
 
 @with_langtrace_root_span()
 def example():
@@ -208,10 +208,8 @@ def example():
 - `with_additional_attributes` - this function is designed to enhance the traces by adding custom attributes to the current context. These custom attributes provide extra details about the operations being performed, making it easier to analyze and understand their behavior.
 
 ```python
-from langtrace_python_sdk.utils.with_root_span import (
-    with_langtrace_root_span,
-    with_additional_attributes,
-)
+from langtrace_python_sdk import with_langtrace_root_span, with_additional_attributes
+
 
 @with_additional_attributes({"user.id": "1234", "user.feedback.rating": 1})
 def api_call1():
@@ -277,6 +275,9 @@ Langtrace automatically captures traces from the following vendors:
 
 We welcome contributions to this project. To get started, fork this repository and start developing. To get involved, join our [Discord](https://discord.langtrace.ai) workspace.
 
+If you want to run any of the examples go to `run_example.py` file, you will find `ENABLED_EXAMPLES`. choose the example you want to run and just toggle the flag to `True` and run the file using `python src/run_example.py`
+
+
 ---
 
 ## Security

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/README.md

@@ -153,7 +153,7 @@ langtrace.init(custom_remote_exporter=<your_exporter>, batch=<True or False>)
 - `@with_langtrace_root_span` - this decorator is designed to organize and relate different spans, in a hierarchical manner. When you're performing multiple operations that you want to monitor together as a unit, this function helps by establishing a "parent" (`LangtraceRootSpan` or whatever is passed to `name`) span. Then, any calls to the LLM APIs made within the given function (fn) will be considered "children" of this parent span. This setup is especially useful for tracking the performance or behavior of a group of operations collectively, rather than individually.
 
 ```python
-from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
+from langtrace_python_sdk import with_langtrace_root_span
 
 @with_langtrace_root_span()
 def example():
@@ -168,10 +168,8 @@ def example():
 - `with_additional_attributes` - this function is designed to enhance the traces by adding custom attributes to the current context. These custom attributes provide extra details about the operations being performed, making it easier to analyze and understand their behavior.
 
 ```python
-from langtrace_python_sdk.utils.with_root_span import (
-    with_langtrace_root_span,
-    with_additional_attributes,
-)
+from langtrace_python_sdk import with_langtrace_root_span, with_additional_attributes
+
 
 @with_additional_attributes({"user.id": "1234", "user.feedback.rating": 1})
 def api_call1():
@@ -237,6 +235,9 @@ Langtrace automatically captures traces from the following vendors:
 
 We welcome contributions to this project. To get started, fork this repository and start developing. To get involved, join our [Discord](https://discord.langtrace.ai) workspace.
 
+If you want to run any of the examples go to `run_example.py` file, you will find `ENABLED_EXAMPLES`. choose the example you want to run and just toggle the flag to `True` and run the file using `python src/run_example.py`
+
+
 ---
 
 ## Security

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/pyproject.toml

@@ -18,7 +18,7 @@ classifiers=[
     "Operating System :: OS Independent",
 ]
 dependencies = [
-  'trace-attributes==4.0.5',
+  'trace-attributes>=5.0.0,<6.0.0',
   'opentelemetry-api>=1.25.0',
   'opentelemetry-sdk>=1.25.0',
   'opentelemetry-instrumentation>=0.46b0',

langtrace_python_sdk-2.1.14/src/examples/__init__.py

@@ -0,0 +1 @@
+

langtrace_python_sdk-2.1.14/src/examples/anthropic_example/__init__.py

@@ -0,0 +1,8 @@
+from examples.anthropic_example.completion import messages_create
+from langtrace_python_sdk import with_langtrace_root_span
+
+
+class AnthropicRunner:
+    @with_langtrace_root_span("Anthropic")
+    def run(self):
+        messages_create()

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/anthropic_example/completion.py

@@ -7,12 +7,11 @@ from langtrace_python_sdk import langtrace, with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init()
 
 
 @with_langtrace_root_span("messages_create")
 def messages_create():
-
     client = anthropic.Anthropic()
 
     message = client.messages.create(

langtrace_python_sdk-2.1.14/src/examples/chroma_example/__init__.py

@@ -0,0 +1,8 @@
+from examples.chroma_example.basic import basic
+from langtrace_python_sdk import with_langtrace_root_span
+
+
+class ChromaRunner:
+    @with_langtrace_root_span("Chroma")
+    def run(self):
+        basic()

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/chroma_example/basic.py

@@ -7,7 +7,7 @@ from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init(write_spans_to_console=False)
 
 
 @with_langtrace_root_span()

langtrace_python_sdk-2.1.14/src/examples/cohere_example/__init__.py

@@ -0,0 +1,17 @@
+from examples.cohere_example.chat import chat_comp
+from examples.cohere_example.chat_stream import chat_stream
+from examples.cohere_example.tools import tool_calling
+from examples.cohere_example.embed import embed
+from examples.cohere_example.rerank import rerank
+from langtrace_python_sdk import with_langtrace_root_span
+
+
+class CohereRunner:
+
+    @with_langtrace_root_span("Cohere")
+    def run(self):
+        chat_comp()
+        chat_stream()
+        tool_calling()
+        embed()
+        rerank()

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/cohere_example/chat.py

@@ -7,7 +7,7 @@ from langtrace_python_sdk import langtrace
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init()
 
 
 co = cohere.Client()

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/cohere_example/chat_stream.py

@@ -5,7 +5,7 @@ from langtrace_python_sdk import langtrace
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init()
 
 
 co = cohere.Client()

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/cohere_example/embed.py

@@ -7,7 +7,7 @@ from langtrace_python_sdk import langtrace
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init()
 
 
 co = cohere.Client()

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/cohere_example/rerank.py

@@ -7,7 +7,7 @@ from langtrace_python_sdk import langtrace
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init()
 
 
 co = cohere.Client()

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/cohere_example/tools.py

@@ -7,7 +7,7 @@ from langtrace_python_sdk import langtrace
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init()
 
 co = cohere.Client()
 

langtrace_python_sdk-2.1.14/src/examples/fastapi_example/__init__.py

@@ -0,0 +1,6 @@
+from .basic_route import root
+
+
+class FastAPIRunner:
+    def run(self):
+        root()

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/fastapi_example/basic_route.py

@@ -7,7 +7,9 @@ from langchain_openai import ChatOpenAI, OpenAIEmbeddings
 from openai import OpenAI
 
 from langtrace_python_sdk import langtrace
+from dotenv import load_dotenv
 
+load_dotenv()
 langtrace.init(write_spans_to_console=True)
 app = FastAPI()
 client = OpenAI()

langtrace_python_sdk-2.1.14/src/examples/langchain_example/__init__.py

@@ -0,0 +1,10 @@
+from .basic import basic_app, rag, load_and_split
+from langtrace_python_sdk import with_langtrace_root_span
+
+
+class LangChainRunner:
+    @with_langtrace_root_span("LangChain")
+    def run(self):
+        basic_app()
+        rag()
+        load_and_split()

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/langchain_example/basic.py

@@ -18,7 +18,7 @@ _ = load_dotenv(find_dotenv())
 langtrace.init()
 
 
-@with_additional_attributes({"user.id": "1234", "user.feedback.rating": 1})
+# @with_additional_attributes({"user.id": "1234", "user.feedback.rating": 1})
 def api_call_1():
     llm = ChatOpenAI()
     prompt = ChatPromptTemplate.from_messages(
@@ -33,7 +33,7 @@ def api_call_1():
     print(res)
 
 
-@with_additional_attributes({"user.id": "37373", "user.feedback.rating": 1})
+# @with_additional_attributes({"user.id": "37373", "user.feedback.rating": 1})
 def api_call_2():
     llm = ChatOpenAI()
     prompt = ChatPromptTemplate.from_messages(
@@ -49,7 +49,7 @@ def api_call_2():
 
 
 @with_langtrace_root_span()
-def basic():
+def basic_app():
     api_call_1()
     api_call_2()
 

langtrace_python_sdk-2.1.14/src/examples/llamaindex_example/__init__.py

@@ -0,0 +1,11 @@
+from langtrace_python_sdk import with_langtrace_root_span
+
+
+class LlamaIndexRunner:
+    @with_langtrace_root_span("LlamaIndex")
+    def run(self):
+        from .basic import basic_app
+        from .agent import main
+
+        basic_app()
+        main()

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/llamaindex_example/agent.py

@@ -12,7 +12,7 @@ import nest_asyncio
 
 nest_asyncio.apply()
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init(write_spans_to_console=False)
 
 
 def multiply(a: int, b: int) -> int:
@@ -82,5 +82,6 @@ class YourOpenAIAgent:
 
 
 # agent = YourOpenAIAgent(tools=[multiply_tool, add_tool])
-llm = OpenAI(model="gpt-3.5-turbo-0613")
-agent = OpenAIAgent.from_tools([multiply_tool, add_tool], llm=llm, verbose=True)
+def main():
+    llm = OpenAI(model="gpt-3.5-turbo-0613")
+    agent = OpenAIAgent.from_tools([multiply_tool, add_tool], llm=llm, verbose=True)

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/llamaindex_example/basic.py

@@ -7,11 +7,11 @@ from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 _ = load_dotenv(find_dotenv())
 
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init(write_spans_to_console=False)
 
 
 @with_langtrace_root_span()
-def basic():
+def basic_app():
     documents = SimpleDirectoryReader(
         "src/examples/llamaindex_example/data"
     ).load_data()

langtrace_python_sdk-2.1.14/src/examples/openai_example/__init__.py

@@ -0,0 +1,21 @@
+from langtrace_python_sdk import with_langtrace_root_span
+
+
+class OpenAIRunner:
+    @with_langtrace_root_span("OpenAI")
+    def run(self):
+        import asyncio
+        from .async_tool_calling_nonstreaming import run_conversation
+        from .async_tool_calling_streaming import (
+            run_conversation as run_conversation_streaming,
+        )
+        from .chat_completion import chat_completion as chat_completion_example
+        from .embeddings_create import embeddings_create as embeddings_create_example
+        from .function_calling import function_calling as function_example
+        from .images_edit import image_edit
+
+        asyncio.run(run_conversation())
+        asyncio.run(run_conversation_streaming())
+        chat_completion_example()
+        embeddings_create_example()
+        function_example()

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/openai_example/async_tool_calling_nonstreaming.py

@@ -3,13 +3,12 @@ import json
 from dotenv import find_dotenv, load_dotenv
 from openai import AsyncOpenAI
 
-from langtrace_python_sdk import langtrace
+from langtrace_python_sdk import langtrace, with_langtrace_root_span
 
-# from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init()
 
 client = AsyncOpenAI()
 
@@ -30,6 +29,7 @@ def get_current_weather(location, unit="fahrenheit"):
         return json.dumps({"location": location, "temperature": "unknown"})
 
 
+@with_langtrace_root_span("Run Conversation")
 async def run_conversation():
     # Step 1: send the conversation and available functions to the model
     messages = [

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/openai_example/async_tool_calling_streaming.py

@@ -3,7 +3,7 @@ import json
 from dotenv import find_dotenv, load_dotenv
 from openai import AsyncOpenAI
 
-from langtrace_python_sdk import langtrace
+from langtrace_python_sdk import langtrace, with_langtrace_root_span
 
 # from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
@@ -42,6 +42,7 @@ def get_current_time(location):
         return json.dumps({"location": location, "time": "unknown"})
 
 
+@with_langtrace_root_span("Run Conversation Streaming")
 async def run_conversation():
     # Step 1: send the conversation and available functions to the model
     messages = [

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/openai_example/chat_completion.py

@@ -9,11 +9,11 @@ from langtrace_python_sdk.utils.with_root_span import (
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init(write_spans_to_console=False)
 client = OpenAI()
 
 
-# @with_additional_attributes({"user.id": "1234", "user.feedback.rating": 1})
+@with_additional_attributes({"user.id": "1234", "user.feedback.rating": 1})
 def api():
     response = client.chat.completions.create(
         model="gpt-4",
@@ -27,7 +27,7 @@ def api():
     return response
 
 
-# @with_langtrace_root_span()
+@with_langtrace_root_span("Chat Completion")
 def chat_completion():
     response = api()
     # print(response)

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/openai_example/embeddings_create.py

@@ -6,12 +6,12 @@ from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init(write_spans_to_console=False)
 
 client = OpenAI()
 
 
-@with_langtrace_root_span()
+@with_langtrace_root_span("Embeddings Create")
 def embeddings_create():
     result = client.embeddings.create(
         model="text-embedding-ada-002",

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/openai_example/function_calling.py

@@ -3,11 +3,11 @@
 from dotenv import find_dotenv, load_dotenv
 from openai import OpenAI
 
-from langtrace_python_sdk import langtrace
+from langtrace_python_sdk import langtrace, with_langtrace_root_span
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init(write_spans_to_console=False)
 
 
 client = OpenAI()
@@ -34,6 +34,7 @@ student_custom_functions = [
 ]
 
 
+@with_langtrace_root_span("Function Calling")
 def function_calling():
     response = client.chat.completions.create(
         model="gpt-3.5-turbo",

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/openai_example/images_edit.py

@@ -8,7 +8,7 @@ from langtrace_python_sdk import langtrace
 
 _ = load_dotenv(find_dotenv())
 
-langtrace.init(write_spans_to_console=True)
+langtrace.init(write_spans_to_console=False)
 
 
 client = OpenAI()
@@ -36,5 +36,5 @@ def image_edit():
     print(response)
 
 
-image_edit()
+# image_edit()
 # convert_to_rgba()

langtrace_python_sdk-2.1.14/src/examples/pinecone_example/__init__.py

@@ -0,0 +1,9 @@
+from langtrace_python_sdk import with_langtrace_root_span
+
+
+class PineconeRunner:
+    @with_langtrace_root_span("Pinecone")
+    def run(self):
+        from .basic import basic as basic_app
+
+        basic_app()

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/pinecone_example/basic.py

@@ -6,7 +6,11 @@ from dotenv import find_dotenv, load_dotenv
 from openai import OpenAI
 from pinecone import Pinecone, ServerlessSpec
 
-from langtrace_python_sdk import langtrace, with_langtrace_root_span
+from langtrace_python_sdk import (
+    langtrace,
+    with_langtrace_root_span,
+    with_additional_attributes,
+)
 from langtrace_python_sdk.utils.with_root_span import SendUserFeedback
 
 _ = load_dotenv(find_dotenv())
@@ -28,7 +32,8 @@ def create_index():
     )
 
 
-@with_langtrace_root_span()
+@with_additional_attributes({"db.embedding_model": "text-embedding-ada-002"})
+@with_langtrace_root_span("Pinecone Basic")
 def basic(span_id=None, trace_id=None):
 
     result = client.embeddings.create(

langtrace_python_sdk-2.1.14/src/examples/qdrant_example/__init__.py

@@ -0,0 +1,8 @@
+from langtrace_python_sdk import with_langtrace_root_span
+from .basic import basic as basic_app
+
+
+class QdrantRunner:
+    @with_langtrace_root_span("Qdrant")
+    def run(self):
+        basic_app()

langtrace_python_sdk-2.1.14/src/examples/weaviate_example/__init__.py

@@ -0,0 +1,26 @@
+from .query_text import (
+    create,
+    insert,
+    query_data_bm25,
+    query_fetch_object_by_id,
+    query_fetch_objects,
+    query_hybrid,
+    query_near_object,
+    query_data_near_text,
+    query_data_near_vector,
+)
+from langtrace_python_sdk import with_langtrace_root_span
+
+
+class WeaviateRunner:
+    @with_langtrace_root_span("Weaviate")
+    def run(self):
+        create()
+        insert()
+        query_data_bm25()
+        # query_fetch_object_by_id()
+        # query_fetch_objects()
+        # query_hybrid()
+        # query_near_object()
+        # query_data_near_text()
+        # query_data_near_vector()

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/examples/weaviate_example/query_text.py

@@ -21,7 +21,9 @@ from weaviate.collections.classes.grpc import Move
 
 import langtrace_python_sdk.langtrace as langtrace
 from langtrace_python_sdk import with_langtrace_root_span
+from dotenv import load_dotenv
 
+load_dotenv()
 # Set these environment variables
 WCS_DEMO_URL = os.environ["WCS_DEMO_URL"]
 WCS_DEMO_RO_KEY = os.environ["WCS_DEMO_RO_KEY"]
@@ -37,17 +39,19 @@ client = weaviate.connect_to_wcs(
 langtrace.init()
 
 
-@with_langtrace_root_span()
+@with_langtrace_root_span("create")
 def create():
-    questions = client.collections.create(
-        name="Question",
-        # Set the vectorizer to "text2vec-openai" to use the OpenAI API for vector-related operations
-        vectorizer_config=wvc.config.Configure.Vectorizer.text2vec_openai(),
-        # Ensure the `generative-openai` module is used for generative queries
-        generative_config=wvc.config.Configure.Generative.openai(),
-    )
+    if not client.collections.get("Question"):
+        questions = client.collections.create(
+            name="Question",
+            # Set the vectorizer to "text2vec-openai" to use the OpenAI API for vector-related operations
+            vectorizer_config=wvc.config.Configure.Vectorizer.text2vec_openai(),
+            # Ensure the `generative-openai` module is used for generative queries
+            generative_config=wvc.config.Configure.Generative.openai(),
+        )
 
 
+@with_langtrace_root_span("insert")
 def insert():
     resp = requests.get(
         "https://raw.githubusercontent.com/weaviate-tutorials/quickstart/main/data/jeopardy_tiny.json"

{langtrace_python_sdk-2.1.13 → langtrace_python_sdk-2.1.14}/src/langtrace_python_sdk/__init__.py

@@ -15,13 +15,18 @@ limitations under the License.
 """
 
 from langtrace_python_sdk import langtrace
-from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
+from langtrace_python_sdk.utils.with_root_span import (
+    with_langtrace_root_span,
+    with_additional_attributes,
+)
+
 from langtrace_python_sdk.utils.prompt_registry import get_prompt_from_registry
 from langtrace_python_sdk.utils.with_root_span import SendUserFeedback
 
 __all__ = [
     "langtrace",
     "with_langtrace_root_span",
+    "with_additional_attributes",
     "get_prompt_from_registry",
     "SendUserFeedback",
 ]

langtrace_python_sdk-2.1.14/src/langtrace_python_sdk/version.py

@@ -0,0 +1 @@
+__version__ = "2.1.14"