Skip to content
New issue

Have a question about this project? # for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “#”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? # to your account

[OPIK-569] [SDK] Implement integration with aisuite #870

Merged
merged 8 commits into from
Dec 13, 2024
Merged

Conversation

japdubengsub
Copy link
Contributor

@japdubengsub japdubengsub commented Dec 11, 2024

Details

Integration with AI Suite has been added.

Usage example:

import aisuite
from opik.integrations.aisuite import track_aisuite

client = aisuite.Client()
wrapped_client = track_aisuite(
    aisuite_client=client,
)
messages = [
    {"role": "system", "content": "You are a helpful assistant."},
    {"role": "user", "content": "Tell a fact"},
]

result = wrapped_client.chat.completions.create(
    model="anthropic:claude-3-5-sonnet-latest",
    messages=messages,
    max_tokens=10,
)

Testing

Integration test were added.

@japdubengsub japdubengsub changed the title OPIK-569 [SDK] Implement integration with aisuite [wip] [OPIK-569] [SDK] Implement integration with aisuite Dec 11, 2024
@japdubengsub japdubengsub marked this pull request as ready for review December 13, 2024 13:24
@japdubengsub japdubengsub requested review from a team as code owners December 13, 2024 13:24
@alexkuzmik alexkuzmik merged commit 0d63dbf into main Dec 13, 2024
26 checks passed
@alexkuzmik alexkuzmik deleted the OPIK-569 branch December 13, 2024 13:41
# for free to join this conversation on GitHub. Already have an account? # to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants