Skip to content
New issue

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

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

Already on GitHub? Sign in to your account

MLflow Gateway models #507

Merged
merged 8 commits into from
Oct 25, 2023
Merged

MLflow Gateway models #507

merged 8 commits into from
Oct 25, 2023

Conversation

fedefernandez
Copy link
Contributor

@fedefernandez fedefernandez commented Oct 25, 2023

MLflow Gateway integration

This PR consists of three parts. The client and its models are defined in a new integration module. Then, the example uses the new integration.

Related to #479

MLflow Gateway Models

File: integrations/mlflow/src/commonMain/kotlin/com/xebia/functional/xef/mlflow/models.kt

Includes the models for interacting with MLflow Gateway:

  • Routes definitions: Include routes information like URI, model names, model type, etc...
  • Chat, completion, and embeddings: Requests and responses for interacting with the exposed models

MLflow Gateway Client

File: integrations/mlflow/src/commonMain/kotlin/com/xebia/functional/xef/mlflow/MlflowClient.kt

Client with some methods for interacting with the gateway and its models.

  • searchRoutes
  • getRoute
  • prompt
  • chat
  • embeddings

MLflow Gateway Example

File: examples/kotlin/src/main/kotlin/com/xebia/functional/xef/conversation/mlflow/Example.kt

Example inspired by the example shown in the following talk by Corey Zumar and Ben Wilson

Copy link
Contributor

@raulraja raulraja left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

thank you @fedefernandez !

@fedefernandez fedefernandez merged commit 3986df1 into main Oct 25, 2023
5 checks passed
@fedefernandez fedefernandez deleted the feature/mlflow-models branch October 25, 2023 10:35
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants