langtrace-python-sdk 1.0.10__tar.gz → 1.0.11__tar.gz

{langtrace-python-sdk-1.0.10 → langtrace-python-sdk-1.0.11}/PKG-INFO

@@ -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 → langtrace-python-sdk-1.0.11}/setup.py

@@ -4,7 +4,7 @@ with open('requirements.txt') as f:
 
 setup(
     name='langtrace-python-sdk',  # Choose a unique name for PyPI
-    version='1.0.10',
+    version='1.0.11',
     author='Ali Waleed',
     author_email='ali@scale3labs.com',
     description='LangTrace - Python SDK',
@@ -12,7 +12,7 @@ setup(
     long_description_content_type='text/markdown',
     url='https://github.com/Scale3-Labs/langtrace-python-sdk',  # Project home page
     package_dir={'': 'src'},
-    packages=find_packages(where='src', exclude=['examples*']),
+    packages=find_packages(where='src'),
     install_requires=required,
     python_requires='>=3.6',
     classifiers=[

langtrace-python-sdk-1.0.11/src/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)

langtrace-python-sdk-1.0.11/src/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 → langtrace-python-sdk-1.0.11}/src/langtrace_python_sdk.egg-info/PKG-INFO

@@ -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 → langtrace-python-sdk-1.0.11}/src/langtrace_python_sdk.egg-info/SOURCES.txt

@@ -2,6 +2,7 @@ LICENSE
 README.md
 setup.py
 src/examples/__init__.py
+src/examples/setup.py
 src/examples/chroma_example/__init__.py
 src/examples/chroma_example/basic.py
 src/examples/langchain_example/__init__.py
@@ -12,6 +13,7 @@ src/examples/llamaindex_example/basic.py
 src/examples/openai/__init__.py
 src/examples/openai/chat_completion.py
 src/examples/openai/embeddings_create.py
+src/examples/openai/function_calling.py
 src/examples/openai/images_generate.py
 src/examples/pinecone_example/__init__.py
 src/examples/pinecone_example/basic.py

{langtrace-python-sdk-1.0.10 → langtrace-python-sdk-1.0.11}/src/langtrace_python_sdk.egg-info/top_level.txt

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