Skip to content

Commit

Permalink
fix(pinecone): filter argument may be dict (#984)
Browse files Browse the repository at this point in the history
  • Loading branch information
nirga authored May 6, 2024
1 parent 4949f91 commit 9c3323a
Showing 1 changed file with 7 additions and 1 deletion.
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""OpenTelemetry Pinecone instrumentation"""

import logging
import json
from opentelemetry.instrumentation.pinecone.config import Config
from opentelemetry.instrumentation.pinecone.utils import dont_throw
import pinecone
Expand Down Expand Up @@ -85,7 +86,12 @@ def _set_query_input_attributes(span, kwargs):
_set_span_attribute(span, "pinecone.query.queries", kwargs.get("queries"))
_set_span_attribute(span, "pinecone.query.top_k", kwargs.get("top_k"))
_set_span_attribute(span, "pinecone.query.namespace", kwargs.get("namespace"))
_set_span_attribute(span, "pinecone.query.filter", kwargs.get("filter"))
if isinstance(kwargs.get("filter"), dict):
_set_span_attribute(
span, "pinecone.query.filter", json.dumps(kwargs.get("filter"))
)
else:
_set_span_attribute(span, "pinecone.query.filter", kwargs.get("filter"))
_set_span_attribute(
span, "pinecone.query.include_values", kwargs.get("include_values")
)
Expand Down

0 comments on commit 9c3323a

Please sign in to comment.