langtrace-python-sdk 1.0.10__py3-none-any.whl → 1.0.11__py3-none-any.whl

examples/chroma_example/basic.py

@@ -0,0 +1,27 @@
+import chromadb
+from chromadb.utils import embedding_functions
+from dotenv import find_dotenv, load_dotenv
+
+from examples.setup import setup_instrumentation
+from instrumentation.with_root_span import with_langtrace_root_span
+
+_ = load_dotenv(find_dotenv())
+
+setup_instrumentation()
+
+
+@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

@@ -0,0 +1,65 @@
+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 examples.setup import setup_instrumentation
+from instrumentation.with_root_span import with_langtrace_root_span
+
+_ = load_dotenv(find_dotenv())
+
+setup_instrumentation()
+
+
+@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

@@ -0,0 +1,85 @@
+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 examples.setup import setup_instrumentation
+from instrumentation.with_root_span import with_langtrace_root_span
+
+_ = load_dotenv(find_dotenv())
+
+setup_instrumentation()
+
+
+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

@@ -0,0 +1,19 @@
+from dotenv import find_dotenv, load_dotenv
+from llama_index.core import SimpleDirectoryReader, VectorStoreIndex
+
+from examples.setup import setup_instrumentation
+from instrumentation.with_root_span import with_langtrace_root_span
+
+_ = load_dotenv(find_dotenv())
+
+setup_instrumentation()
+
+
+@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

@@ -0,0 +1,39 @@
+from dotenv import find_dotenv, load_dotenv
+from openai import OpenAI
+
+from examples.setup import setup_instrumentation
+from instrumentation.with_root_span import with_langtrace_root_span
+
+_ = load_dotenv(find_dotenv())
+
+setup_instrumentation()
+
+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.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))

examples/openai/embeddings_create.py

@@ -0,0 +1,19 @@
+from dotenv import find_dotenv, load_dotenv
+from openai import OpenAI
+
+from examples.setup import setup_instrumentation
+from instrumentation.with_root_span import with_langtrace_root_span
+
+_ = load_dotenv(find_dotenv())
+
+setup_instrumentation()
+
+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

@@ -0,0 +1,75 @@
+import json
+
+from dotenv import find_dotenv, load_dotenv
+from openai import OpenAI
+
+from examples.setup import setup_instrumentation
+from instrumentation.with_root_span import with_langtrace_root_span
+
+_ = load_dotenv(find_dotenv())
+
+setup_instrumentation()
+
+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

@@ -0,0 +1,20 @@
+from dotenv import find_dotenv, load_dotenv
+from openai import OpenAI
+
+from examples.setup import setup_instrumentation
+from instrumentation.with_root_span import with_langtrace_root_span
+
+_ = load_dotenv(find_dotenv())
+
+setup_instrumentation()
+
+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

@@ -0,0 +1,32 @@
+from dotenv import find_dotenv, load_dotenv
+from openai import OpenAI
+from pinecone import Pinecone
+
+from examples.setup import setup_instrumentation
+from instrumentation.with_root_span import with_langtrace_root_span
+
+_ = load_dotenv(find_dotenv())
+
+setup_instrumentation()
+
+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)

examples/setup.py

@@ -0,0 +1,50 @@
+
+from opentelemetry import trace
+from opentelemetry.sdk.trace import TracerProvider
+from opentelemetry.sdk.trace.export import (ConsoleSpanExporter,
+                                            SimpleSpanProcessor)
+
+from instrumentation.chroma.instrumentation import ChromaInstrumentation
+from instrumentation.langchain.instrumentation import LangchainInstrumentation
+from instrumentation.langchain_community.instrumentation import \
+    LangchainCommunityInstrumentation
+from instrumentation.langchain_core.instrumentation import \
+    LangchainCoreInstrumentation
+from instrumentation.llamaindex.instrumentation import \
+    LlamaindexInstrumentation
+from instrumentation.openai.instrumentation import OpenAIInstrumentation
+from instrumentation.pinecone.instrumentation import PineconeInstrumentation
+
+
+def setup_instrumentation():
+
+    # Set up OpenTelemetry tracing
+    tracer_provider = TracerProvider()
+
+    # Use the ConsoleSpanExporter to print traces to the console
+    console_exporter = ConsoleSpanExporter()
+    tracer_provider.add_span_processor(SimpleSpanProcessor(console_exporter))
+
+    # Initialize tracer
+    trace.set_tracer_provider(tracer_provider)
+
+    # Initialize and enable your custom OpenAI instrumentation
+    # Create an instance of OpenAIInstrumentation
+    openai_instrumentation = OpenAIInstrumentation()
+    pinecone_instrumentation = PineconeInstrumentation()
+    llamaindex_instrumentation = LlamaindexInstrumentation()
+    chroma_instrumentation = ChromaInstrumentation()
+    langchain_instrumentation = LangchainInstrumentation()
+    langchain_core_instrumentation = LangchainCoreInstrumentation()
+    langchain_community_instrumentation = LangchainCommunityInstrumentation()
+
+    # Call the instrument method with some arguments
+    openai_instrumentation.instrument()
+    pinecone_instrumentation.instrument()
+    llamaindex_instrumentation.instrument()
+    chroma_instrumentation.instrument()
+    langchain_instrumentation.instrument()
+    langchain_core_instrumentation.instrument()
+    langchain_community_instrumentation.instrument()
+
+    print("setup complete")

{langtrace_python_sdk-1.0.10.dist-info → langtrace_python_sdk-1.0.11.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: langtrace-python-sdk
-Version: 1.0.10
+Version: 1.0.11
 Summary: LangTrace - Python SDK
 Home-page: https://github.com/Scale3-Labs/langtrace-python-sdk
 Author: Ali Waleed

{langtrace_python_sdk-1.0.10.dist-info → langtrace_python_sdk-1.0.11.dist-info}/RECORD

@@ -1,3 +1,19 @@
+examples/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+examples/setup.py,sha256=07Sy36lUuNuPU5QPAs2BEMm-YKSosruzKJPl0QKc_rc,2105
+examples/chroma_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+examples/chroma_example/basic.py,sha256=HD8vGSsvcXN21AuWyUx09CePphXrIp9PSNllueMqvEI,836
+examples/langchain_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+examples/langchain_example/basic.py,sha256=PBYotFDJlkH9y17U8WRbH3FxqgfBPHra2k_C7hHPiMc,1993
+examples/langchain_example/tool.py,sha256=g7mFuXal6TtCvo-KTYqWeSEpuyQMR2LuHF6WQdKRwbw,2494
+examples/llamaindex_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+examples/llamaindex_example/basic.py,sha256=b1y1VnYE7eaeSXLk9rJ4EZApCj-tMPPMTY-fc_M7X7c,620
+examples/openai/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+examples/openai/chat_completion.py,sha256=3ojkccgntJp_IXnNb7f7w9sXPtj_mDNOE7IyLB6SsS0,1300
+examples/openai/embeddings_create.py,sha256=01ApMJ2F_8-IWanDAD8SWlrcjg8b3mJzaddsGhRvrck,457
+examples/openai/function_calling.py,sha256=MzxNNvPNdUWOw8Ugs_gMGuwyQN4yPJnja46rE_8W_ZY,2475
+examples/openai/images_generate.py,sha256=GD2Bj902xqS0pfZr0pZtPXCGHEHWZM_CjGJ0DdoP_pY,444
+examples/pinecone_example/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+examples/pinecone_example/basic.py,sha256=lF8TcDvW1v6j-QKvl3B9r1t0ShnvCcOgJ5F74qy6VO8,793
 instrumentation/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 instrumentation/constants.py,sha256=YdC62dsYpbbBdMHhfUbaK-cbDM4w4eau0ClUdWVmjmU,336
 instrumentation/with_root_span.py,sha256=CRie2ljHhnHN8bUGDwBM-F18-c6xyoI_238KP8BEO-U,969
@@ -30,8 +46,8 @@ instrumentation/pinecone/apis.py,sha256=XpKNUfyzEE3HkBN10Qv1w_t1PT-J39pHlotrdU-w
 instrumentation/pinecone/instrumentation.py,sha256=yfOxKkMtW6GEUQ0E9AWSBdaa07MHzV3o6Q09cAvoWIU,1708
 instrumentation/pinecone/patch.py,sha256=fr07o97CqGc8sUEyMtSiT6watZiTPStRPOrxOzhJGLo,1840
 instrumentation/pinecone/lib/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-langtrace_python_sdk-1.0.10.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-langtrace_python_sdk-1.0.10.dist-info/METADATA,sha256=PD7RbdnqjA8lCTuz7rc89PgW6VcrpMxV_DntIHak4bY,6149
-langtrace_python_sdk-1.0.10.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
-langtrace_python_sdk-1.0.10.dist-info/top_level.txt,sha256=mdFAULSZuqUiDveRElCIPMvwAkRAYXP4bm_dEI4A96Q,16
-langtrace_python_sdk-1.0.10.dist-info/RECORD,,
+langtrace_python_sdk-1.0.11.dist-info/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+langtrace_python_sdk-1.0.11.dist-info/METADATA,sha256=Zgg1c6Ykg62bBICe3rPmogg4GWjTe-WwtnGWQyVJ_mA,6149
+langtrace_python_sdk-1.0.11.dist-info/WHEEL,sha256=oiQVh_5PnQM0E3gPdiz09WCNmwiHDMaGer_elqB3coM,92
+langtrace_python_sdk-1.0.11.dist-info/top_level.txt,sha256=FJhLokAk3p9qB-lzJKTfQ1CtAXoxvyL8zkfffVrFSdI,25
+langtrace_python_sdk-1.0.11.dist-info/RECORD,,

{langtrace_python_sdk-1.0.10.dist-info → langtrace_python_sdk-1.0.11.dist-info}/top_level.txt

@@ -1 +1,2 @@
+examples
 instrumentation