chore: cleanup dependencies

requirements.txt

@@ -1,260 +1 @@
-aiohttp==3.9.2
-aiosignal==1.3.1
-argcomplete==3.2.2
-async-timeout==4.0.2
-attrs==22.1.0
-awsebcli==3.20.3
-backoff==2.2.1
-bcrypt==4.0.0
-black==24.3.0
-blessed==1.19.1
-blinker==1.6.2
-botocore==1.23.54
-cached-property==1.5.2
-cachetools==5.3.1
-cement==2.8.2
-certifi==2023.7.22
-cffi==1.15.1
-charset-normalizer==2.1.1
-click==8.1.7
-colorama==0.4.3
-cryptography==42.0.4
-dataclasses-json==0.5.7
-datamodel-code-generator==0.25.4
-Deprecated==1.2.14
-discord==2.1.0
-discord.py==2.1.0
-dnspython==2.3.0
-docker==4.4.4
-docker-compose==1.25.5
-dockerpty==0.4.1
-docopt==0.6.2
-email-validator==2.1.0.post1
-Flask==2.3.3
-Flask-Cors==4.0.0
-frozenlist==1.3.3
-future==0.18.3
-genson==1.2.2
-google-api-core==2.11.1
-google-api-python-client==2.99.0
-google-auth==2.23.0
-google-auth-httplib2==0.1.1
-google-auth-oauthlib==1.1.0
-google-cloud-core==2.3.3
-google-cloud-storage==2.10.0
-google-crc32c==1.5.0
-google-resumable-media==2.6.0
-google-search-results==2.4.2
-googleapis-common-protos==1.60.0
-grpcio==1.59.2
-grpcio-health-checking==1.59.2
-gspread==5.11.2
-httplib2==0.22.0
-idna==3.4
-importlib-metadata==6.11.0
-inflect==5.6.2
-isort==5.13.2
-itsdangerous==2.1.2
-Jinja2==3.1.3
-jmespath==0.10.0
-jsonschema==3.2.0
-langchain==0.1.11
-MarkupSafe==2.1.3
-marshmallow==3.19.0
-marshmallow-enum==1.5.1
-multidict==6.0.2
-mypy-extensions==1.0.0
-numpy==1.23.5
-oauthlib==3.2.2
-openai==0.27.4
-openapi-schema-pydantic==1.2.4
-opentelemetry-api==1.22.0
-opentelemetry-distro==0.43b0
-opentelemetry-exporter-otlp-proto-common==1.22.0
-opentelemetry-exporter-otlp-proto-grpc==1.22.0
-opentelemetry-instrumentation==0.43b0
-opentelemetry-proto==1.22.0
-opentelemetry-sdk==1.22.0
-opentelemetry-semantic-conventions==0.43b0
-packaging==23.0
-pandas==1.5.2
-paramiko==3.4.0
-pathspec==0.9.0
-platformdirs==4.2.0
-protobuf==4.24.3
-psutil==5.9.6
-psycopg==3.1.8
-psycopg2-binary==2.9.6
-pyasn1==0.5.0
-pyasn1-modules==0.3.0
-pycparser==2.21
-pydantic==1.10.7
-pymongo==4.6.3
-PyNaCl==1.5.0
-pyparsing==3.1.1
-pyrsistent==0.18.1
-python-dateutil==2.8.2
-python-dotenv==1.0.0
-python-json-logger==2.0.7
-pytube==12.1.2
-pytz==2022.6
-PyYAML==6.0.1
-requests==2.31.0
-requests-oauthlib==1.3.1
-rsa==4.9
-semantic-version==2.8.5
-six==1.16.0
-SQLAlchemy==1.4.47
-tenacity==8.2.2
-termcolor==1.1.0
-texttable==1.6.4
-toml==0.10.2
-tomli==2.0.1
-tqdm==4.65.0
-typing-inspect==0.8.0
-typing_extensions==4.5.0
-uritemplate==4.1.1
-urllib3==1.25.11
-wcwidth==0.1.9
-websocket-client==0.59.0
-Werkzeug==2.3.8
-wrapt==1.16.0
-yarl==1.8.1
-zipp==3.17.0
-aiohttp==3.9.2
-aiosignal==1.3.1
-argcomplete==3.2.2
-async-timeout==4.0.2
-attrs==22.1.0
-awsebcli==3.20.3
-backoff==2.2.1
-bcrypt==4.0.0
-black==24.3.0
-blessed==1.19.1
-blinker==1.6.2
-botocore==1.23.54
-cached-property==1.5.2
-cachetools==5.3.1
-cement==2.8.2
-certifi==2023.7.22
-cffi==1.15.1
-charset-normalizer==2.1.1
-click==8.1.7
-colorama==0.4.3
-cryptography==42.0.4
-dataclasses-json==0.5.7
-datamodel-code-generator==0.25.4
-Deprecated==1.2.14
-discord==2.1.0
-discord.py==2.1.0
-dnspython==2.3.0
-docker==4.4.4
-docker-compose==1.25.5
-dockerpty==0.4.1
-docopt==0.6.2
-docutils==0.20.1
-email-validator==2.1.0.post1
-Flask==2.3.3
-Flask-Cors==4.0.0
-frozenlist==1.3.3
-future==0.18.3
-genson==1.2.2
-google-api-core==2.11.1
-google-api-python-client==2.99.0
-google-auth==2.23.0
-google-auth-httplib2==0.1.1
-google-auth-oauthlib==1.1.0
-google-cloud-core==2.3.3
-google-cloud-storage==2.10.0
-google-crc32c==1.5.0
-google-resumable-media==2.6.0
-google-search-results==2.4.2
-googleapis-common-protos==1.60.0
-grpcio==1.59.2
-grpcio-health-checking==1.59.2
-gspread==5.11.2
-httplib2==0.22.0
-idna==3.4
-importlib-metadata==6.11.0
-inflect==5.6.2
-isort==5.13.2
-itsdangerous==2.1.2
-jaraco.classes==3.3.1
-Jinja2==3.1.3
-jmespath==0.10.0
-jsonschema==3.2.0
-keyring==24.3.0
-langchain==0.1.11
-markdown-it-py==3.0.0
-MarkupSafe==2.1.3
-marshmallow==3.19.0
-marshmallow-enum==1.5.1
-mdurl==0.1.2
-more-itertools==10.2.0
-multidict==6.0.2
-mypy-extensions==1.0.0
-nh3==0.2.15
-numpy==1.23.5
-oauthlib==3.2.2
-openai==0.27.4
-openapi-schema-pydantic==1.2.4
-opentelemetry-api==1.22.0
-opentelemetry-distro==0.43b0
-opentelemetry-exporter-otlp-proto-common==1.22.0
-opentelemetry-exporter-otlp-proto-grpc==1.22.0
-opentelemetry-instrumentation==0.43b0
-opentelemetry-proto==1.22.0
-opentelemetry-sdk==1.22.0
-opentelemetry-semantic-conventions==0.43b0
-packaging==23.0
-pandas==1.5.2
-paramiko==3.4.0
-pathspec==0.9.0
-pkginfo==1.9.6
-platformdirs==4.2.0
-protobuf==4.24.3
-psutil==5.9.6
-psycopg==3.1.8
-psycopg2-binary==2.9.6
-pyasn1==0.5.0
-pyasn1-modules==0.3.0
-pycparser==2.21
-pydantic==1.10.7
-Pygments==2.17.2
-pymongo==4.6.3
-PyNaCl==1.5.0
-pyparsing==3.1.1
-pyrsistent==0.18.1
-python-dateutil==2.8.2
-python-dotenv==1.0.0
-python-json-logger==2.0.7
-pytube==12.1.2
-pytz==2022.6
-PyYAML==6.0.1
-readme-renderer==42.0
-requests==2.31.0
-requests-oauthlib==1.3.1
-requests-toolbelt==1.0.0
-rfc3986==2.0.0
-rich==13.7.0
-rsa==4.9
-semantic-version==2.8.5
-six==1.16.0
-SQLAlchemy==1.4.47
-tenacity==8.2.2
-termcolor==1.1.0
-texttable==1.6.4
-toml==0.10.2
-tomli==2.0.1
-tqdm==4.65.0
-twine==5.0.0
-typing-inspect==0.8.0
-typing_extensions==4.5.0
-uritemplate==4.1.1
-urllib3==1.26.18
-wcwidth==0.1.9
-websocket-client==0.59.0
-Werkzeug==2.3.8
-wrapt==1.16.0
-yarl==1.8.1
-zipp==3.17.0
+pydantic==2.7.2

setup.py

@@ -1,24 +1,24 @@
 from setuptools import find_packages, setup
 
 setup(
-    name='trace-attributes',  # Choose a unique name for PyPI
-    version='4.0.3',
-    author='Karthik Kalyanaraman',
-    author_email='[email protected]',
-    description='LangTrace - Trace Attributes',
+    name="trace-attributes",  # Choose a unique name for PyPI
+    version="4.0.4",
+    author="Karthik Kalyanaraman",
+    author_email="[email protected]",
+    description="LangTrace - Trace Attributes",
     long_description="LangTrace - Trace Attributes",
-    long_description_content_type='text/markdown',
-    url='https://github.com/Scale3-Labs/langtrace-trace-attributes',  # Project home page
-    package_dir={'': 'src/python'},
-    packages=find_packages(where='src/python'),
+    long_description_content_type="text/markdown",
+    url="https://github.com/Scale3-Labs/langtrace-trace-attributes",  # Project home page
+    package_dir={"": "src/python"},
+    packages=find_packages(where="src/python"),
     install_requires=[
-        'pydantic>=1.8',  # Example dependency, adjust according to your project's needs
+        "pydantic>=1.8",  # Example dependency, adjust according to your project's needs
     ],
-    python_requires='>=3.6',
+    python_requires=">=3.6",
     classifiers=[
-        'Programming Language :: Python :: 3',
-        'License :: OSI Approved :: MIT License',
-        'Operating System :: OS Independent',
+        "Programming Language :: Python :: 3",
+        "License :: OSI Approved :: MIT License",
+        "Operating System :: OS Independent",
     ],
     include_package_data=False,  # To include non-code files specified in MANIFEST.in
 )

src/python/__init__.py

@@ -1 +1 @@
-# __init__.py
+# __init__.py

src/python/langtrace/__init__.py

@@ -1 +1 @@
-# __init__.py
+# __init__.py

src/python/langtrace/trace_attributes/__init__.py

@@ -71,5 +71,13 @@ class WeaviateMethods(Enum):
 
 
 # Export only what you want to be accessible directly through `import my_package`
-__all__ = ['LLMSpanAttributes', 'DatabaseSpanAttributes', 'FrameworkSpanAttributes', 'Event',
-           'OpenAIMethods', 'ChromaDBMethods', 'PineconeMethods', 'WeaviateMethods']
+__all__ = [
+    "LLMSpanAttributes",
+    "DatabaseSpanAttributes",
+    "FrameworkSpanAttributes",
+    "Event",
+    "OpenAIMethods",
+    "ChromaDBMethods",
+    "PineconeMethods",
+    "WeaviateMethods",
+]

src/python/langtrace/trace_attributes/models/__init__.py

@@ -1 +1 @@
-# __init__.py
+# __init__.py

src/python/langtrace/trace_attributes/models/database_span_attributes.py

@@ -6,28 +6,27 @@
 
 from typing import Optional
 
-from pydantic import BaseModel, Extra, Field
+from pydantic import BaseModel, ConfigDict, Field
 
 
 class DatabaseSpanAttributes(BaseModel):
-    class Config:
-        extra = Extra.forbid
+    model_config = ConfigDict(extra="forbid")
 
-    langtrace_service_name: str = Field(..., alias='langtrace.service.name')
-    langtrace_service_type: str = Field(..., alias='langtrace.service.type')
+    langtrace_service_name: str = Field(..., alias="langtrace.service.name")
+    langtrace_service_type: str = Field(..., alias="langtrace.service.type")
     langtrace_service_version: Optional[str] = Field(
-        None, alias='langtrace.service.version'
+        None, alias="langtrace.service.version"
     )
-    langtrace_sdk_name: str = Field(..., alias='langtrace.sdk.name')
-    langtrace_version: str = Field(..., alias='langtrace.version')
-    server_address: Optional[str] = Field(None, alias='server.address')
-    db_query: str = Field(..., alias='db.query')
-    db_response: Optional[str] = Field(None, alias='db.response')
-    db_operation: Optional[str] = Field(None, alias='db.operation')
-    db_system: str = Field(..., alias='db.system')
-    db_namespace: Optional[str] = Field(None, alias='db.namespace')
-    db_index: Optional[str] = Field(None, alias='db.index')
-    db_collection_name: Optional[str] = Field(None, alias='db.collection.name')
-    db_top_k: Optional[float] = Field(None, alias='db.top_k')
-    db_embedding_model: Optional[str] = Field(None, alias='db.embedding_model')
-    langtrace_testId: Optional[str] = Field(None, alias='langtrace.testId')
+    langtrace_sdk_name: str = Field(..., alias="langtrace.sdk.name")
+    langtrace_version: str = Field(..., alias="langtrace.version")
+    server_address: Optional[str] = Field(None, alias="server.address")
+    db_query: str = Field(..., alias="db.query")
+    db_response: Optional[str] = Field(None, alias="db.response")
+    db_operation: Optional[str] = Field(None, alias="db.operation")
+    db_system: str = Field(..., alias="db.system")
+    db_namespace: Optional[str] = Field(None, alias="db.namespace")
+    db_index: Optional[str] = Field(None, alias="db.index")
+    db_collection_name: Optional[str] = Field(None, alias="db.collection.name")
+    db_top_k: Optional[float] = Field(None, alias="db.top_k")
+    db_embedding_model: Optional[str] = Field(None, alias="db.embedding_model")
+    langtrace_testId: Optional[str] = Field(None, alias="langtrace.testId")