langtrace-python-sdk 1.1.29__py3-none-any.whl → 1.2.5__py3-none-any.whl

langtrace_python_sdk/instrumentation/anthropic/patch.py

@@ -9,7 +9,6 @@ from opentelemetry.trace.status import Status, StatusCode
 
 from langtrace_python_sdk.constants.instrumentation.anthropic import APIS
 from langtrace_python_sdk.constants.instrumentation.common import SERVICE_PROVIDERS
-from langtrace_python_sdk.utils.llm import estimate_tokens
 
 
 def messages_create(original_method, version, tracer):

langtrace_python_sdk/instrumentation/openai/patch.py

@@ -97,7 +97,6 @@ def chat_completions_create(original_method, version, tracer):
             "langtrace.version": "1.0.0",
             "url.full": base_url,
             "llm.api": APIS["CHAT_COMPLETION"]["ENDPOINT"],
-            "llm.model": kwargs.get("model"),
             "llm.prompts": json.dumps(kwargs.get("messages", [])),
             "llm.stream": kwargs.get("stream"),
         }
@@ -126,28 +125,34 @@ def chat_completions_create(original_method, version, tracer):
             # Attempt to call the original method
             result = original_method(*args, **kwargs)
             if kwargs.get("stream") is False:
+                span.set_attribute("llm.model", result.model)
                 if hasattr(result, "choices") and result.choices is not None:
                     responses = [
                         {
-                            "message": (
-                                choice.message.content
-                                if choice.message and choice.message.content
-                                else (
-                                    choice.message.function_call.arguments
-                                    if choice.message
-                                    and choice.message.function_call.arguments
-                                    else ""
-                                )
-                            ),
-                            **(
-                                {
-                                    "content_filter_results": choice[
-                                        "content_filter_results"
-                                    ]
-                                }
-                                if "content_filter_results" in choice
-                                else {}
-                            ),
+                            "message": {
+                                    "role": choice.message.role
+                                    if choice.message and choice.message.role
+                                    else "assistant",
+                                    "content": (
+                                    choice.message.content
+                                    if choice.message and choice.message.content
+                                    else (
+                                        choice.message.function_call.arguments
+                                        if choice.message
+                                        and choice.message.function_call.arguments
+                                        else ""
+                                    )
+                                ),
+                                **(
+                                    {
+                                        "content_filter_results": choice[
+                                            "content_filter_results"
+                                        ]
+                                    }
+                                    if "content_filter_results" in choice
+                                    else {}
+                                ),
+                            }
                         }
                         for choice in result.choices
                     ]
@@ -167,8 +172,8 @@ def chat_completions_create(original_method, version, tracer):
                     usage = result.usage
                     if usage is not None:
                         usage_dict = {
-                            "prompt_tokens": result.usage.prompt_tokens,
-                            "completion_tokens": usage.completion_tokens,
+                            "input_tokens": result.usage.prompt_tokens,
+                            "output_tokens": usage.completion_tokens,
                             "total_tokens": usage.total_tokens,
                         }
                         span.set_attribute("llm.token.counts", json.dumps(usage_dict))
@@ -244,8 +249,8 @@ def chat_completions_create(original_method, version, tracer):
                 "llm.token.counts",
                 json.dumps(
                     {
-                        "prompt_tokens": prompt_tokens,
-                        "completion_tokens": completion_tokens,
+                        "input_tokens": prompt_tokens,
+                        "output_tokens": completion_tokens,
                         "total_tokens": prompt_tokens + completion_tokens,
                     }
                 ),

langtrace_python_sdk/version.py

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

langtrace_python_sdk-1.2.5.dist-info/METADATA

@@ -0,0 +1,63 @@
+Metadata-Version: 2.3
+Name: langtrace-python-sdk
+Version: 1.2.5
+Summary: Python SDK for LangTrace
+Project-URL: Homepage, https://github.com/Scale3-Labs/langtrace-python-sdk
+Author-email: Scale3 Labs <engineering@scale3labs.com>
+License: AGPL-3.0-or-later
+License-File: LICENSE
+Classifier: License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+
+# What is Langtrace?
+
+Langtrace stands as a developer-centric, open-source solution, fully compatible with OpenTelemetry. It enables developers to effortlessly trace, monitor, and debug their LLM applications, offering robust support for automatic instrumentation.
+
+## Supported LLM Modules
+
+Langtrace supports a comprehensive range of LLMs, VectorDBs, and frameworks, ensuring wide coverage for your development needs:
+
+### LLMs
+
+1. OpenAI
+2. Anthropic
+3. Azure OpenAI
+
+### VectorDBs
+
+1. Pinecone
+2. Chromadb
+
+### Frameworks
+
+1. LangChain
+2. LlamaIndex
+3. Haystack
+
+We are actively working to extend our support to additional libraries!
+
+## Getting Started
+
+To begin utilizing Langtrace, follow these straightforward steps:
+
+1. Install the package using `pip install langtrace-python-sdk`.
+2. Incorporate Langtrace into your project with `from langtrace_python_sdk import langtrace`.
+   - This import should precede any other LLM module imports (such as OpenAI, LlamaIndex, etc.) to ensure proper functionality.
+3. Initialize Langtrace by adding `langtrace.init({ write_to_remote_url: false})` to your code.
+4. Congratulations, you've completed the basic setup! You will now begin to see traces from your LLM modules logged directly to the console.
+
+## Exporting Traces to Langtrace
+
+To configure trace exporting, you have two options:
+
+You'll need both a Langtrace `api_key` and a `remote_url`, which can be acquired by logging into your Langtrace account.
+
+1. Direct Initialization: Utilize `langtrace.init(batch=True, api_key=<YOUR_API_KEY>, remote_url=<YOUR_REMOTE_URL>)`.
+2. Environment Variables: Set `API_KEY` and `URL`, then add `LangTrace.init(batch=True)` at the beginning of your file.
+
+## Langtrace Cloud
+
+Currently under development 🚧

{langtrace_python_sdk-1.1.29.dist-info → langtrace_python_sdk-1.2.5.dist-info}/RECORD

@@ -1,22 +1,6 @@
-examples/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/anthropic_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/anthropic_example/completion.py,sha256=IDZVPUpqp31LZtA-wJUBSdG-vEGPtqyOToxo25ifvXU,659
-examples/chroma_example/__init__.py,sha256=Tq6pae7fHA7dwuDslabB5MTNedL21gu2RaZicpxSyLU,25
-examples/chroma_example/basic.py,sha256=pmDTgborQYxMRpcJfyu4XlJDguBcb9FR5gNLksrzEZk,883
-examples/langchain_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/langchain_example/basic.py,sha256=iQpOatVVzbJYVoGlDaXlJG5C7uQG-aoePBAWh0Dhi2Y,2073
-examples/langchain_example/tool.py,sha256=8T8_IDbgA58XbsfyH5_xhA8ZKQfyfyFxF8wor-PsRjA,2556
-examples/llamaindex_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/llamaindex_example/basic.py,sha256=ewjyxzx4ntuID6D5Myo1UFj_F6KGd5yT0yQntCPQ3Rw,688
-examples/openai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/openai/chat_completion.py,sha256=QmmnKFZTrYCySbQHvWkIn1UDF4Ii_DsM-WPaRrnH1F8,1331
-examples/openai/embeddings_create.py,sha256=AhDNAqg-WzRYLJAE_b2RKGjuVCh4aZSU7MxcZv2kCHQ,518
-examples/openai/function_calling.py,sha256=Y5IZ2eb78dHaWRqOb3k6K0tn6-Zca_TLsV8qwuu1iYA,2369
-examples/openai/images_generate.py,sha256=ZioxTuHKE_yYlhpESqXKVzdkiwdegkmLVB7N8T2LU00,506
-examples/pinecone_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-examples/pinecone_example/basic.py,sha256=8Tpf0hYMgJdjVpISu15BW7IpjrnwZXrBuCGoC9xE7Fg,927
 langtrace_python_sdk/__init__.py,sha256=SlHg447-nQBbw8exRNJP_OyHUZ39Sldb7aaQ35hIRm8,262
 langtrace_python_sdk/langtrace.py,sha256=XS4lhorNLPgsdKB2hTFsHR31_kUgr9cHZTc4dLgJs_g,3251
+langtrace_python_sdk/version.py,sha256=jBmZf3HLbiQlWiolOsAA6J5-BbxXD2bqFqEqDH3lfqo,22
 langtrace_python_sdk/constants/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/constants/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/constants/instrumentation/anthropic.py,sha256=YX3llt3zwDY6XrYk3CB8WEVqgrzRXEw_ffyk56JoF3k,126
@@ -29,7 +13,7 @@ langtrace_python_sdk/extensions/langtrace_exporter.py,sha256=6dMS0MjYCmZbuj3CMb9
 langtrace_python_sdk/instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/anthropic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/anthropic/instrumentation.py,sha256=1shNkDE7Yb-JFVlEZHMcDbJ6zW8SkSFdxf_yFo9wTNA,1101
-langtrace_python_sdk/instrumentation/anthropic/patch.py,sha256=oc_HCzZbYGa4CNnZAfKmmrEPtNlVZwuvpB8V1_Lcohg,6579
+langtrace_python_sdk/instrumentation/anthropic/patch.py,sha256=w4Jcz5McG5NHXlr1IrwJmqj2P2d5vt7uxkpa2OWpbXU,6520
 langtrace_python_sdk/instrumentation/chroma/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/chroma/instrumentation.py,sha256=fSR1ZjoViwjbPYkUWJnRpFwWtwERyf4UqnMiDl8eGmY,1223
 langtrace_python_sdk/instrumentation/chroma/patch.py,sha256=NGHZR5TGaL0JgZX184L8Nt5U3hpDdD-htHUkMgQuT5A,1999
@@ -47,15 +31,14 @@ langtrace_python_sdk/instrumentation/llamaindex/instrumentation.py,sha256=udDxk8
 langtrace_python_sdk/instrumentation/llamaindex/patch.py,sha256=BJyX9AFiDvCZ3EjgU3roSdhb4CNZdKfajPogvBv33vc,1739
 langtrace_python_sdk/instrumentation/openai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/openai/instrumentation.py,sha256=6RmbONTwgr8_do92l3ll7-ia4maXMxQVCwPBRRa-3NA,1358
-langtrace_python_sdk/instrumentation/openai/patch.py,sha256=eB6neGUy79UOBljCSklflwJ2-ILXwNHOGKLmrWV9_hs,13866
+langtrace_python_sdk/instrumentation/openai/patch.py,sha256=FdHW7JEcizQkl3qg2bq7SuLNB4jWJqn64cXnMIg2pno,14219
 langtrace_python_sdk/instrumentation/pinecone/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/instrumentation/pinecone/instrumentation.py,sha256=PBkCMiaDr-WXkGfHt9iXtYqyl7a-xIO52pRmUFuMURo,1726
 langtrace_python_sdk/instrumentation/pinecone/patch.py,sha256=uDKUBjyOVCDL44YxXVR80VSZGgg-ZNCcQulBUwrgYPk,1896
 langtrace_python_sdk/utils/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 langtrace_python_sdk/utils/llm.py,sha256=4z2e-md_ELXCEuOIRVWracR6qH2pmsOxCqpkuF9_3Nw,1589
 langtrace_python_sdk/utils/with_root_span.py,sha256=LgFVwHq7KZ6sj2d783NZXr2fFWqI1rqVWSjJR68Ad1E,1044
-langtrace_python_sdk-1.1.29.dist-info/LICENSE,sha256=VD-pauwiiia-Xi2zgKvalKRIFSJJjqRCQw6aIpK2T9U,33892
-langtrace_python_sdk-1.1.29.dist-info/METADATA,sha256=bOrPWAachBC1rAkDeUd6XSSI5UmpinEqe5-3wTAVhpk,8131
-langtrace_python_sdk-1.1.29.dist-info/WHEEL,sha256=GJ7t_kWBFywbagK5eo9IoUwLW6oyOeTKmQ-9iHFVNxQ,92
-langtrace_python_sdk-1.1.29.dist-info/top_level.txt,sha256=rwET8cHATJvDBti7Dm6pwWIrNAT8DvqCXgRPWdu7_Hs,30
-langtrace_python_sdk-1.1.29.dist-info/RECORD,,
+langtrace_python_sdk-1.2.5.dist-info/METADATA,sha256=c15XCywpwvDGdX-252msSpUxP59N4T9HRI2tFVzm5Es,2283
+langtrace_python_sdk-1.2.5.dist-info/WHEEL,sha256=uNdcs2TADwSd5pVaP0Z_kcjcvvTUklh2S7bxZMF8Uj0,87
+langtrace_python_sdk-1.2.5.dist-info/licenses/LICENSE,sha256=VD-pauwiiia-Xi2zgKvalKRIFSJJjqRCQw6aIpK2T9U,33892
+langtrace_python_sdk-1.2.5.dist-info/RECORD,,

{langtrace_python_sdk-1.1.29.dist-info → langtrace_python_sdk-1.2.5.dist-info}/WHEEL

@@ -1,5 +1,4 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.43.0)
+Generator: hatchling 1.22.4
 Root-Is-Purelib: true
 Tag: py3-none-any
-

examples/anthropic_example/completion.py

@@ -1,27 +0,0 @@
-"""Example of using the anthropic API to create a message."""
-
-import anthropic
-from dotenv import find_dotenv, load_dotenv
-
-from langtrace_python_sdk import langtrace
-
-_ = load_dotenv(find_dotenv())
-
-langtrace.init(batch=False, log_spans_to_console=True, write_to_remote_url=False)
-
-
-def messages_create():
-
-    client = anthropic.Anthropic()
-
-    message = client.messages.create(
-        model="claude-3-opus-20240229",
-        max_tokens=1000,
-        temperature=0.0,
-        system="Respond only in Yoda-speak.",
-        messages=[{"role": "user", "content": "How are you today?"}],
-        stream=True,
-    )
-
-    for response in message:
-        pass

examples/chroma_example/__init__.py

@@ -1 +0,0 @@
-from .basic import basic

examples/chroma_example/basic.py

@@ -1,25 +0,0 @@
-import chromadb
-from chromadb.utils import embedding_functions
-from dotenv import find_dotenv, load_dotenv
-
-from langtrace_python_sdk import langtrace
-from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
-
-_ = load_dotenv(find_dotenv())
-
-langtrace.init(batch=False, log_spans_to_console=True, write_to_remote_url=False)
-
-
-@with_langtrace_root_span()
-def basic():
-    chroma_client = chromadb.Client()
-    embedder = embedding_functions.DefaultEmbeddingFunction()
-    collection = chroma_client.create_collection(
-        name="my6_collection", embedding_function=embedder
-    )
-    collection.add(
-        documents=["This is a document", "This is another document"],
-        metadatas=[{"source": "my_source"}, {"source": "my_source"}],
-        ids=["id1", "id2"],
-    )
-    results = collection.query(query_texts=["This is a query document"], n_results=2)

examples/langchain_example/basic.py

@@ -1,66 +0,0 @@
-from dotenv import find_dotenv, load_dotenv
-from langchain.text_splitter import RecursiveCharacterTextSplitter
-from langchain_community.document_loaders import PyPDFLoader
-from langchain_community.vectorstores.faiss import FAISS
-from langchain_core.output_parsers import StrOutputParser
-from langchain_core.prompts.chat import ChatPromptTemplate
-from langchain_core.runnables import RunnablePassthrough
-from langchain_openai import ChatOpenAI, OpenAIEmbeddings
-
-from langtrace_python_sdk import langtrace
-from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
-
-_ = load_dotenv(find_dotenv())
-
-langtrace.init(batch=True, log_spans_to_console=True, write_to_remote_url=False)
-
-
-@with_langtrace_root_span()
-def basic():
-    llm = ChatOpenAI()
-    prompt = ChatPromptTemplate.from_messages(
-        [
-            ("system", "You are world class technical documentation writer."),
-            ("user", "{input}"),
-        ]
-    )
-    output_parser = StrOutputParser()
-    chain = prompt | llm | output_parser
-    res = chain.invoke({"input": "how can langsmith help with testing?"})
-    print(res)
-
-
-@with_langtrace_root_span()
-def rag():
-    vectorstore = FAISS.from_texts(
-        ["harrison worked at kensho"], embedding=OpenAIEmbeddings()
-    )
-    retriever = vectorstore.as_retriever()
-
-    template = """Answer the question based only on the following context:{context}
-
-        Question: {question}
-    """
-    prompt = ChatPromptTemplate.from_template(template)
-
-    model = ChatOpenAI()
-
-    chain = (
-        {"context": retriever, "question": RunnablePassthrough()}
-        | prompt
-        | model
-        | StrOutputParser()
-    )
-
-    res = chain.invoke("where did harrison work?")
-    # print(res)
-
-
-@with_langtrace_root_span()
-def load_and_split():
-    url = "https://www.w3.org/WAI/ER/tests/xhtml/testfiles/resources/pdf/dummy.pdf"
-    loader = PyPDFLoader(url)
-    data = loader.load()
-    text_splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=0)
-    docs = text_splitter.split_documents(data)
-    # print(docs)

examples/langchain_example/tool.py

@@ -1,85 +0,0 @@
-from dotenv import find_dotenv, load_dotenv
-from langchain import hub
-from langchain.agents import AgentExecutor, create_openai_functions_agent
-from langchain.chains import LLMMathChain
-from langchain_core.pydantic_v1 import BaseModel, Field
-from langchain_core.tools import Tool
-from langchain_openai import ChatOpenAI
-
-from langtrace_python_sdk import langtrace
-from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
-
-_ = load_dotenv(find_dotenv())
-
-langtrace.init(batch=True, log_spans_to_console=True, write_to_remote_url=False)
-
-
-llm = ChatOpenAI(temperature=0, model="gpt-4")
-llm_math_chain = LLMMathChain.from_llm(llm=llm, verbose=True)
-
-primes = {998: 7901, 999: 7907, 1000: 7919}
-
-
-class CalculatorInput(BaseModel):
-    question: str = Field()
-
-
-class PrimeInput(BaseModel):
-    n: int = Field()
-
-
-def is_prime(n: int) -> bool:
-    if n <= 1 or (n % 2 == 0 and n > 2):
-        return False
-    for i in range(3, int(n**0.5) + 1, 2):
-        if n % i == 0:
-            return False
-    return True
-
-
-def get_prime(n: int, primes: dict = primes) -> str:
-    return str(primes.get(int(n)))
-
-
-async def aget_prime(n: int, primes: dict = primes) -> str:
-    return str(primes.get(int(n)))
-
-
-@with_langtrace_root_span()
-def tool_example():
-
-    tools = [
-        Tool(
-            name="GetPrime",
-            func=get_prime,
-            description="A tool that returns the `n`th prime number",
-            args_schema=PrimeInput,
-            coroutine=aget_prime,
-        ),
-        Tool.from_function(
-            func=llm_math_chain.run,
-            name="Calculator",
-            description="Useful for when you need to compute mathematical expressions",
-            args_schema=CalculatorInput,
-            coroutine=llm_math_chain.arun,
-        ),
-    ]
-
-    prompt = hub.pull("hwchase17/openai-functions-agent")
-
-    agent = create_openai_functions_agent(llm, tools, prompt)
-
-    agent_executor = AgentExecutor(agent=agent, tools=tools, verbose=True)
-
-    question = "What is the product of the 998th, 999th and 1000th prime numbers?"
-
-    for step in agent_executor.iter({"input": question}):
-        if output := step.get("intermediate_step"):
-            action, value = output[0]
-            if action.tool == "GetPrime":
-                print(f"Checking whether {value} is prime...")
-                assert is_prime(int(value))
-            # Ask user if they want to continue
-            _continue = input("Should the agent continue (Y/n)?:\n") or "Y"
-            if _continue.lower() != "y":
-                break

examples/llamaindex_example/basic.py

@@ -1,21 +0,0 @@
-from dotenv import find_dotenv, load_dotenv
-from llama_index.core import SimpleDirectoryReader, VectorStoreIndex
-
-from langtrace_python_sdk import langtrace
-from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
-
-_ = load_dotenv(find_dotenv())
-
-
-langtrace.init(batch=True, log_spans_to_console=True, write_to_remote_url=False)
-
-
-@with_langtrace_root_span()
-def basic():
-    documents = SimpleDirectoryReader(
-        "src/examples/llamaindex_example/data"
-    ).load_data()
-    index = VectorStoreIndex.from_documents(documents)
-    query_engine = index.as_query_engine()
-    response = query_engine.query("What did the author do in college?")
-    print(response)

examples/openai/chat_completion.py

@@ -1,39 +0,0 @@
-from dotenv import find_dotenv, load_dotenv
-from openai import OpenAI
-
-from langtrace_python_sdk import langtrace
-from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
-
-_ = load_dotenv(find_dotenv())
-
-langtrace.init(batch=False, log_spans_to_console=True, write_to_remote_url=False)
-
-client = OpenAI()
-
-
-@with_langtrace_root_span()
-def chat_completion():
-    response = client.chat.completions.create(
-        model="gpt-4",
-        messages=[{"role": "user", "content": "Say this is a test three times"}],
-        stream=True,
-    )
-    # print(stream)
-    # stream = client.chat.completions.create(
-    #     model="gpt-4",
-    #     messages=[{"role": "user", "content": "Say this is a test three times"}, {"role": "assistant", "content": "This is a test. This is a test. This is a test"},
-    #               {"role": "user", "content": "Say this is a mock 4 times"}],
-    #     stream=False,
-    # )
-
-    # result = []
-    # for chunk in response:
-    #     if chunk.choices[0].delta.content is not None:
-    #         content = [
-    #             choice.delta.content if choice.delta and
-    #             choice.delta.content else ""
-    #             for choice in chunk.choices]
-    #         result.append(
-    #             content[0] if len(content) > 0 else "")
-
-    # print("".join(result))

examples/openai/embeddings_create.py

@@ -1,18 +0,0 @@
-from dotenv import find_dotenv, load_dotenv
-from openai import OpenAI
-
-from langtrace_python_sdk import langtrace
-from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
-
-_ = load_dotenv(find_dotenv())
-
-langtrace.init(batch=True, log_spans_to_console=True, write_to_remote_url=False)
-client = OpenAI()
-
-
-@with_langtrace_root_span()
-def embeddings_create():
-    result = client.embeddings.create(
-        model="text-embedding-ada-002",
-        input="Once upon a time, there was a frog.",
-    )

examples/openai/function_calling.py

@@ -1,66 +0,0 @@
-import json
-
-from dotenv import find_dotenv, load_dotenv
-from openai import OpenAI
-
-from langtrace_python_sdk import langtrace
-from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
-
-_ = load_dotenv(find_dotenv())
-
-langtrace.init(batch=True, log_spans_to_console=True, write_to_remote_url=False)
-
-client = OpenAI()
-
-
-student_custom_functions = [
-    {
-        "name": "extract_student_info",
-        "description": "Get the student information from the body of the input text",
-        "parameters": {
-            "type": "object",
-            "properties": {
-                "name": {"type": "string", "description": "Name of the person"},
-                "major": {"type": "string", "description": "Major subject."},
-                "school": {"type": "string", "description": "The university name."},
-                "grades": {"type": "integer", "description": "GPA of the student."},
-                "club": {
-                    "type": "string",
-                    "description": "School club for extracurricular activities. ",
-                },
-            },
-        },
-    }
-]
-
-
-@with_langtrace_root_span()
-def function_calling():
-    response = client.chat.completions.create(
-        model="gpt-3.5-turbo",
-        messages=[
-            {
-                "role": "user",
-                "content": "David Nguyen is a sophomore majoring in computer science at Stanford University. He is Asian American and has a 3.8 GPA. David is known for his programming skills and is an active member of the university's Robotics Club. He hopes to pursue a career in artificial intelligence after graduating.",
-            }
-        ],
-        functions=student_custom_functions,
-        function_call="auto",
-        stream=False,
-    )
-
-    # result = []
-    # for chunk in response:
-    #     if chunk.choices[0].delta.function_call is not None:
-    #         content = [
-    #             choice.delta.function_call.arguments if choice.delta.function_call and
-    #             choice.delta.function_call.arguments else ""
-    #             for choice in chunk.choices]
-    #         result.append(
-    #             content[0] if len(content) > 0 else "")
-
-    # print("".join(result))
-
-    # Loading the response as a JSON object
-    json_response = json.loads(response.choices[0].message.function_call.arguments)
-    print(json_response)

examples/openai/images_generate.py

@@ -1,20 +0,0 @@
-from dotenv import find_dotenv, load_dotenv
-from openai import OpenAI
-
-from langtrace_python_sdk import langtrace
-from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
-
-_ = load_dotenv(find_dotenv())
-
-langtrace.init(batch=True, log_spans_to_console=True, write_to_remote_url=False)
-
-client = OpenAI()
-
-
-@with_langtrace_root_span()
-def images_generate():
-    result = client.images.generate(
-        model="dall-e-3",
-        prompt="A cute baby sea otter",
-    )
-    print(result)

examples/pinecone_example/basic.py

@@ -1,36 +0,0 @@
-"""
-This example demonstrates how to use Pinecone with Langtrace.
-"""
-
-from dotenv import find_dotenv, load_dotenv
-from openai import OpenAI
-from pinecone import Pinecone
-
-from langtrace_python_sdk import langtrace
-from langtrace_python_sdk.utils.with_root_span import with_langtrace_root_span
-
-_ = load_dotenv(find_dotenv())
-
-langtrace.init(batch=True, log_spans_to_console=True, write_to_remote_url=False)
-
-client = OpenAI()
-pinecone = Pinecone()
-
-
-@with_langtrace_root_span()
-def basic():
-    result = client.embeddings.create(
-        model="text-embedding-ada-002",
-        input="Some random text string goes here",
-        encoding_format="float",
-    )
-
-    embedding = result.data[0].embedding
-
-    unique_id = "randomid"
-    data_to_upsert = {"id": unique_id, "values": embedding}
-
-    index = pinecone.Index("test-index")
-    index.upsert(vectors=[data_to_upsert])
-
-    resp = index.query(vector=embedding, top_k=1)

langtrace_python_sdk-1.1.29.dist-info/METADATA

@@ -1,220 +0,0 @@
-Metadata-Version: 2.1
-Name: langtrace-python-sdk
-Version: 1.1.29
-Summary: Python SDK for LangTrace
-Home-page: https://github.com/Scale3-Labs/langtrace-python-sdk
-Author: Scale3 Labs
-Author-email: engineering@scale3labs.com
-Maintainer: ['Ali Waleed', 'Darshit Suratwala', 'Dylan Zuber', 'Karthik Kalyanaraman', 'Obinna Okafor', 'Rohit Kadhe', 'Yemi Adejumobi']
-License: AGPL-3.0-or-later
-Classifier: Programming Language :: Python :: 3
-Classifier: License :: OSI Approved :: GNU Affero General Public License v3 or later (AGPLv3+)
-Classifier: Operating System :: OS Independent
-Requires-Python: >=3.6
-Description-Content-Type: text/markdown
-License-File: LICENSE
-Requires-Dist: bcrypt ==4.1.2
-Requires-Dist: aiohttp ==3.9.3
-Requires-Dist: aiosignal ==1.3.1
-Requires-Dist: annotated-types ==0.6.0
-Requires-Dist: anthropic ==0.19.1
-Requires-Dist: anyio ==4.3.0
-Requires-Dist: asgiref ==3.7.2
-Requires-Dist: async-timeout ==4.0.3
-Requires-Dist: attrs ==23.2.0
-Requires-Dist: backoff ==2.2.1
-Requires-Dist: beautifulsoup4 ==4.12.3
-Requires-Dist: bs4 ==0.0.2
-Requires-Dist: build ==1.0.3
-Requires-Dist: cachetools ==5.3.3
-Requires-Dist: certifi ==2024.2.2
-Requires-Dist: charset-normalizer ==3.3.2
-Requires-Dist: chroma-hnswlib ==0.7.3
-Requires-Dist: chromadb ==0.4.23
-Requires-Dist: click ==8.1.7
-Requires-Dist: coloredlogs ==15.0.1
-Requires-Dist: dataclasses-json ==0.6.4
-Requires-Dist: Deprecated ==1.2.14
-Requires-Dist: dirtyjson ==1.0.8
-Requires-Dist: distro ==1.9.0
-Requires-Dist: docarray ==0.40.0
-Requires-Dist: exceptiongroup ==1.2.0
-Requires-Dist: faiss-cpu ==1.7.4
-Requires-Dist: fastapi ==0.110.0
-Requires-Dist: filelock ==3.13.1
-Requires-Dist: flatbuffers ==23.5.26
-Requires-Dist: frozenlist ==1.4.1
-Requires-Dist: fsspec ==2024.2.0
-Requires-Dist: google-auth ==2.28.1
-Requires-Dist: googleapis-common-protos ==1.62.0
-Requires-Dist: greenlet ==3.0.3
-Requires-Dist: grpcio ==1.62.0
-Requires-Dist: h11 ==0.14.0
-Requires-Dist: httpcore ==1.0.4
-Requires-Dist: httptools ==0.6.1
-Requires-Dist: httpx ==0.27.0
-Requires-Dist: huggingface-hub ==0.20.3
-Requires-Dist: humanfriendly ==10.0
-Requires-Dist: idna ==3.6
-Requires-Dist: importlib-metadata ==6.11.0
-Requires-Dist: importlib-resources ==6.1.2
-Requires-Dist: joblib ==1.3.2
-Requires-Dist: jsonpatch ==1.33
-Requires-Dist: jsonpointer ==2.4
-Requires-Dist: kubernetes ==29.0.0
-Requires-Dist: langchain ==0.1.9
-Requires-Dist: langchain-community ==0.0.24
-Requires-Dist: langchain-core ==0.1.27
-Requires-Dist: langchain-openai ==0.0.7
-Requires-Dist: langchainhub ==0.1.14
-Requires-Dist: langsmith ==0.1.9
-Requires-Dist: llama-index ==0.10.13.post1
-Requires-Dist: llama-index-agent-openai ==0.1.5
-Requires-Dist: llama-index-cli ==0.1.5
-Requires-Dist: llama-index-core ==0.10.13
-Requires-Dist: llama-index-embeddings-openai ==0.1.6
-Requires-Dist: llama-index-indices-managed-llama-cloud ==0.1.3
-Requires-Dist: llama-index-legacy ==0.9.48
-Requires-Dist: llama-index-llms-openai ==0.1.6
-Requires-Dist: llama-index-multi-modal-llms-openai ==0.1.4
-Requires-Dist: llama-index-program-openai ==0.1.4
-Requires-Dist: llama-index-question-gen-openai ==0.1.3
-Requires-Dist: llama-index-readers-file ==0.1.6
-Requires-Dist: llama-index-readers-llama-parse ==0.1.3
-Requires-Dist: llama-index-vector-stores-chroma ==0.1.4
-Requires-Dist: llama-parse ==0.3.4
-Requires-Dist: llamaindex-py-client ==0.1.13
-Requires-Dist: markdown-it-py ==3.0.0
-Requires-Dist: marshmallow ==3.21.0
-Requires-Dist: mdurl ==0.1.2
-Requires-Dist: mmh3 ==4.1.0
-Requires-Dist: monotonic ==1.6
-Requires-Dist: mpmath ==1.3.0
-Requires-Dist: multidict ==6.0.5
-Requires-Dist: mypy-extensions ==1.0.0
-Requires-Dist: nest-asyncio ==1.6.0
-Requires-Dist: networkx ==3.2.1
-Requires-Dist: nltk ==3.8.1
-Requires-Dist: numexpr ==2.9.0
-Requires-Dist: numpy ==1.26.4
-Requires-Dist: oauthlib ==3.2.2
-Requires-Dist: onnxruntime ==1.17.1
-Requires-Dist: openai ==1.12.0
-Requires-Dist: opentelemetry-api ==1.23.0
-Requires-Dist: opentelemetry-distro ==0.44b0
-Requires-Dist: opentelemetry-exporter-otlp ==1.23.0
-Requires-Dist: opentelemetry-exporter-otlp-proto-common ==1.23.0
-Requires-Dist: opentelemetry-exporter-otlp-proto-grpc ==1.23.0
-Requires-Dist: opentelemetry-exporter-otlp-proto-http ==1.23.0
-Requires-Dist: opentelemetry-instrumentation ==0.44b0
-Requires-Dist: opentelemetry-instrumentation-asgi ==0.44b0
-Requires-Dist: opentelemetry-instrumentation-fastapi ==0.44b0
-Requires-Dist: opentelemetry-proto ==1.23.0
-Requires-Dist: opentelemetry-sdk ==1.23.0
-Requires-Dist: opentelemetry-semantic-conventions ==0.44b0
-Requires-Dist: opentelemetry-util-http ==0.44b0
-Requires-Dist: orjson ==3.9.15
-Requires-Dist: overrides ==7.7.0
-Requires-Dist: packaging ==23.2
-Requires-Dist: pandas ==2.2.1
-Requires-Dist: pillow ==10.2.0
-Requires-Dist: pinecone-client ==3.1.0
-Requires-Dist: posthog ==3.4.2
-Requires-Dist: protobuf ==4.25.3
-Requires-Dist: pulsar-client ==3.4.0
-Requires-Dist: pyasn1 ==0.5.1
-Requires-Dist: pyasn1-modules ==0.3.0
-Requires-Dist: pydantic ==2.6.2
-Requires-Dist: pydantic-core ==2.16.3
-Requires-Dist: Pygments ==2.17.2
-Requires-Dist: PyMuPDF ==1.23.25
-Requires-Dist: PyMuPDFb ==1.23.22
-Requires-Dist: pypdf ==4.0.2
-Requires-Dist: PyPika ==0.48.9
-Requires-Dist: pyproject-hooks ==1.0.0
-Requires-Dist: python-dateutil ==2.8.2
-Requires-Dist: python-dotenv ==1.0.1
-Requires-Dist: pytz ==2024.1
-Requires-Dist: PyYAML ==6.0.1
-Requires-Dist: regex ==2023.12.25
-Requires-Dist: requests ==2.31.0
-Requires-Dist: requests-oauthlib ==1.3.1
-Requires-Dist: rich ==13.7.0
-Requires-Dist: rsa ==4.9
-Requires-Dist: six ==1.16.0
-Requires-Dist: sniffio ==1.3.0
-Requires-Dist: soupsieve ==2.5
-Requires-Dist: SQLAlchemy ==2.0.27
-Requires-Dist: starlette ==0.36.3
-Requires-Dist: sympy ==1.12
-Requires-Dist: tenacity ==8.2.3
-Requires-Dist: tiktoken ==0.6.0
-Requires-Dist: tokenizers ==0.15.2
-Requires-Dist: tomli ==2.0.1
-Requires-Dist: tqdm ==4.66.2
-Requires-Dist: trace-attributes ==1.0.24
-Requires-Dist: typer ==0.9.0
-Requires-Dist: types-requests ==2.31.0.20240218
-Requires-Dist: typing ==3.7.4.3
-Requires-Dist: typing-inspect ==0.9.0
-Requires-Dist: typing-extensions ==4.9.0
-Requires-Dist: tzdata ==2024.1
-Requires-Dist: urllib3 ==2.2.1
-Requires-Dist: uvicorn ==0.27.1
-Requires-Dist: uvloop ==0.19.0
-Requires-Dist: watchfiles ==0.21.0
-Requires-Dist: websocket-client ==1.7.0
-Requires-Dist: websockets ==12.0
-Requires-Dist: wrapt ==1.16.0
-Requires-Dist: yarl ==1.9.4
-Requires-Dist: zipp ==3.17.0
-
-# What is Langtrace?
-
-Langtrace stands as a developer-centric, open-source solution, fully compatible with OpenTelemetry. It enables developers to effortlessly trace, monitor, and debug their LLM applications, offering robust support for automatic instrumentation.
-
-## Supported LLM Modules
-
-Langtrace supports a comprehensive range of LLMs, VectorDBs, and frameworks, ensuring wide coverage for your development needs:
-
-### LLMs
-
-1. OpenAI
-2. Anthropic
-3. Azure OpenAI
-
-### VectorDBs
-
-1. Pinecone
-2. Chromadb
-
-### Frameworks
-
-1. LangChain
-2. LlamaIndex
-3. Haystack
-
-We are actively working to extend our support to additional libraries!
-
-## Getting Started
-
-To begin utilizing Langtrace, follow these straightforward steps:
-
-1. Install the package using `pip install langtrace-python-sdk`.
-2. Incorporate Langtrace into your project with `from langtrace_python_sdk import langtrace`.
-   - This import should precede any other LLM module imports (such as OpenAI, LlamaIndex, etc.) to ensure proper functionality.
-3. Initialize Langtrace by adding `langtrace.init({ write_to_remote_url: false})` to your code.
-4. Congratulations, you've completed the basic setup! You will now begin to see traces from your LLM modules logged directly to the console.
-
-## Exporting Traces to Langtrace
-
-To configure trace exporting, you have two options:
-
-You'll need both a Langtrace `api_key` and a `remote_url`, which can be acquired by logging into your Langtrace account.
-
-1. Direct Initialization: Utilize `langtrace.init(batch=True, api_key=<YOUR_API_KEY>, remote_url=<YOUR_REMOTE_URL>)`.
-2. Environment Variables: Set `API_KEY` and `URL`, then add `LangTrace.init(batch=True)` at the beginning of your file.
-
-## Langtrace Cloud
-
-Currently under development 🚧

langtrace_python_sdk-1.1.29.dist-info/top_level.txt

@@ -1,2 +0,0 @@
-examples
-langtrace_python_sdk