frogml-core 0.0.113__tar.gz → 0.0.114__tar.gz

{frogml_core-0.0.113 → frogml_core-0.0.114}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: frogml-core
-Version: 0.0.113
+Version: 0.0.114
 Summary: frogml Core contains the necessary objects and communication tools for using the Jfrog ml Platform
 License: Apache-2.0
 Keywords: mlops,ml,deployment,serving,model

{frogml_core-0.0.113 → frogml_core-0.0.114}/frogml_core/__init__.py

@@ -1,7 +1,7 @@
 """Top-level package for frogml."""
 
 __author__ = "jfrog"
-__version__ = "0.0.113"
+__version__ = "0.0.114"
 
 from frogml_core.inner.di_configuration import wire_dependencies
 from frogml_core.model.model_version_tracking import (  # noqa: F401,E501

{frogml_core-0.0.113 → frogml_core-0.0.114}/frogml_core/inner/di_configuration/__init__.py

@@ -32,11 +32,8 @@ def wire_dependencies():
         model_management,
         model_version_manager,
         project,
-        prompt_manager,
         system_secret,
         user_application_instance,
-        vector_store,
-        workspace_manager,
         jfrog_gateway,
     )
 
@@ -62,11 +59,8 @@ def wire_dependencies():
             feature_store,
             user_application_instance,
             alerts_registry,
-            workspace_manager,
-            vector_store,
             integration_management,
             system_secret,
-            prompt_manager,
             model_version_manager,
             jfrog_gateway,
         ]

{frogml_core-0.0.113 → frogml_core-0.0.114}/frogml_services_mock/mocks/frogml_mocks.py

@@ -1,6 +1,5 @@
 import dataclasses
 
-from frogml_services_mock.mocks import workspace_manager_service_mock
 from frogml_services_mock.mocks.alert_manager_service_api import (
     AlertManagerServiceApiMock,
 )
@@ -76,7 +75,6 @@ from frogml_services_mock.mocks.model_version_manager_service import (
     ModelVersionManagerServiceMock,
 )
 from frogml_services_mock.mocks.project_manager_service import ProjectManagerServiceMock
-from frogml_services_mock.mocks.prompt_manager_service import PromptManagerServiceMock
 from frogml_services_mock.mocks.repository_service_mock import RepositoryServiceMock
 from frogml_services_mock.mocks.secret_service import SecretServiceMock
 from frogml_services_mock.mocks.self_service_user_service import (
@@ -86,11 +84,6 @@ from frogml_services_mock.mocks.system_secret_service import SystemSecretService
 from frogml_services_mock.mocks.user_application_instance_service_api import (
     UserApplicationInstanceServiceApiMock,
 )
-from frogml_services_mock.mocks.vector_serving_api import VectorServingServiceMock
-from frogml_services_mock.mocks.vectors_management_api import (
-    VectorCollectionManagementServiceMock,
-)
-
 
 @dataclasses.dataclass
 class FrogmlMocks:
@@ -126,13 +119,9 @@ class FrogmlMocks:
     fs_feature_sets_service: FeatureSetServiceMock
     instance_templates_service: InstanceTemplateManagementServiceMock
     alerts_registry_service: AlertsRegistryServiceApiMock
-    workspace_manager_service: workspace_manager_service_mock
-    vector_serving_service: VectorServingServiceMock
-    vector_collection_service: VectorCollectionManagementServiceMock
     execution_management_service: ExecutionManagementServiceMock
     system_secret_service: SystemSecretServiceMock
     integration_management_service: IntegrationManagementServiceMock
-    prompt_manager_service: PromptManagerServiceMock
     model_version_manager_service: ModelVersionManagerServiceMock
     repository_service: RepositoryServiceMock
     location_discovery_service: LocationDiscoveryServiceApiMock

{frogml_core-0.0.113 → frogml_core-0.0.114}/frogml_services_mock/services_mock.py

@@ -107,9 +107,6 @@ from frogml_proto.qwak.offline.serving.v1.offline_serving_async_service_pb2_grpc
 from frogml_proto.qwak.projects.projects_pb2_grpc import (
     add_ProjectsManagementServiceServicer_to_server,
 )
-from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2_grpc import (
-    add_PromptManagerServiceServicer_to_server,
-)
 from frogml_proto.qwak.secret_service.secret_service_pb2_grpc import (
     add_SecretServiceServicer_to_server,
 )
@@ -119,15 +116,6 @@ from frogml_proto.qwak.self_service.user.v1.user_service_pb2_grpc import (
 from frogml_proto.qwak.service_discovery.service_discovery_location_service_pb2_grpc import (
     add_LocationDiscoveryServiceServicer_to_server,
 )
-from frogml_proto.qwak.vectors.v1.collection.collection_service_pb2_grpc import (
-    add_VectorCollectionServiceServicer_to_server,
-)
-from frogml_proto.qwak.vectors.v1.vector_service_pb2_grpc import (
-    add_VectorServiceServicer_to_server,
-)
-from frogml_proto.qwak.workspace.workspace_service_pb2_grpc import (
-    add_WorkspaceManagementServiceServicer_to_server,
-)
 from frogml_services_mock.mocks.alert_manager_service_api import (
     AlertManagerServiceApiMock,
 )
@@ -204,7 +192,6 @@ from frogml_services_mock.mocks.model_version_manager_service import (
     ModelVersionManagerServiceMock,
 )
 from frogml_services_mock.mocks.project_manager_service import ProjectManagerServiceMock
-from frogml_services_mock.mocks.prompt_manager_service import PromptManagerServiceMock
 from frogml_services_mock.mocks.repository_service_mock import RepositoryServiceMock
 from frogml_services_mock.mocks.secret_service import SecretServiceMock
 from frogml_services_mock.mocks.self_service_user_service import (
@@ -214,13 +201,6 @@ from frogml_services_mock.mocks.system_secret_service import SystemSecretService
 from frogml_services_mock.mocks.user_application_instance_service_api import (
     UserApplicationInstanceServiceApiMock,
 )
-from frogml_services_mock.mocks.vector_serving_api import VectorServingServiceMock
-from frogml_services_mock.mocks.vectors_management_api import (
-    VectorCollectionManagementServiceMock,
-)
-from frogml_services_mock.mocks.workspace_manager_service_mock import (
-    WorkspaceManagerServiceMock,
-)
 from frogml_services_mock.utils.service_utils import find_free_port
 
 
@@ -267,8 +247,6 @@ def frogml_container():
         project,
         secret_service,
         user_application_instance,
-        vector_store,
-        workspace_manager,
     )
     from frogml_core.clients.administration import (
         authentication,
@@ -276,9 +254,7 @@ def frogml_container():
         self_service,
     )
     from frogml_core.clients.integration_management import integration_manager_client
-    from frogml_core.clients.prompt_manager import prompt_manager_client
     from frogml_core.clients.system_secret import system_secret_client
-    from frogml_core.vector_store.utils import upsert_utils
 
     container.wire(
         packages=[
@@ -305,12 +281,8 @@ def frogml_container():
             user_application_instance,
             secret_service,
             alerts_registry,
-            workspace_manager,
-            vector_store,
-            upsert_utils,
             integration_manager_client,
             system_secret_client,
-            prompt_manager_client,
             model_version_manager,
             jfrog_gateway,
             location_discovery,
@@ -503,31 +475,11 @@ def attach_servicers(free_port, server):
                 AlertsRegistryServiceApiMock,
                 add_AlertingChannelManagementServiceServicer_to_server,
             ),
-            (
-                "workspace_manager_service",
-                WorkspaceManagerServiceMock,
-                add_WorkspaceManagementServiceServicer_to_server,
-            ),
-            (
-                "vector_serving_service",
-                VectorServingServiceMock,
-                add_VectorServiceServicer_to_server,
-            ),
-            (
-                "vector_collection_service",
-                VectorCollectionManagementServiceMock,
-                add_VectorCollectionServiceServicer_to_server,
-            ),
             (
                 "execution_management_service",
                 ExecutionManagementServiceMock,
                 add_FeatureStoreExecutionServiceServicer_to_server,
             ),
-            (
-                "prompt_manager_service",
-                PromptManagerServiceMock,
-                add_PromptManagerServiceServicer_to_server,
-            ),
             (
                 "model_version_manager_service",
                 ModelVersionManagerServiceMock,

frogml_core-0.0.114/frogml_storage/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.0.114"

{frogml_core-0.0.113 → frogml_core-0.0.114}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "frogml-core"
-version = "0.0.113"
+version = "0.0.114"
 description = "frogml Core contains the necessary objects and communication tools for using the Jfrog ml Platform"
 authors = ["Jfrog"]
 readme = "README.md"

frogml_core-0.0.113/frogml_core/clients/prompt_manager/model_descriptor_mapper.py

@@ -1,196 +0,0 @@
-from typing import List, Optional, Union
-
-from google.protobuf import json_format
-from google.protobuf.struct_pb2 import ListValue, Struct
-
-from frogml_proto.qwak.model_descriptor.open_ai_descriptor_pb2 import (
-    OpenAICapabilities as ProtoOpenAICapabilities,
-)
-from frogml_proto.qwak.model_descriptor.open_ai_descriptor_pb2 import (
-    OpenAIChatAPI as ProtoOpenAIChatAPI,
-)
-from frogml_proto.qwak.model_descriptor.open_ai_descriptor_pb2 import (
-    OpenAIChatModelParams as ProtoOpenAIChatModelParams,
-)
-from frogml_proto.qwak.model_descriptor.open_ai_descriptor_pb2 import (
-    OpenAIModelDescriptor as ProtoOpenAIModelDescriptor,
-)
-from frogml_proto.qwak.model_descriptor.open_ai_descriptor_pb2 import (
-    ToolChoice as ProtoToolChoice,
-)
-from frogml_proto.qwak.model_descriptor.open_ai_descriptor_pb2 import (
-    Tools as ProtoTools,
-)
-from frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import (
-    PromptModelDescriptor as ProtoPromptModelDescriptor,
-)
-from frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import (
-    PromptOpenAIProvider as ProtoPromptOpenAIProvider,
-)
-from frogml_core.exceptions import FrogmlException
-from frogml_core.llmops.generation.chat.openai.types.chat.chat_completion_named_tool_choice_param import (
-    ChatCompletionNamedToolChoiceParam,
-)
-from frogml_core.llmops.generation.chat.openai.types.chat.chat_completion_tool_param import (
-    ChatCompletionToolParam,
-)
-from frogml_core.llmops.model.descriptor import ModelDescriptor, OpenAIChat
-
-
-class ModelDescriptorMapper:
-    @staticmethod
-    def _from_tool_choice(
-        openai_chat_params: ProtoOpenAIChatModelParams,
-    ) -> Optional[Union[str, ChatCompletionNamedToolChoiceParam]]:
-        _tool_choice: Optional[Union[str, ChatCompletionNamedToolChoiceParam]] = None
-        if openai_chat_params.HasField("tool_choice"):
-            if openai_chat_params.tool_choice.WhichOneof("value_type") == "json":
-                _tool_choice = json_format.MessageToDict(
-                    openai_chat_params.tool_choice.json
-                )
-            elif openai_chat_params.tool_choice.WhichOneof("value_type") == "literal":
-                _tool_choice = openai_chat_params.tool_choice.literal
-        return _tool_choice
-
-    @staticmethod
-    def from_openai_chat_capability(
-        model_id: str, openai_chat_params: ProtoOpenAIChatModelParams
-    ) -> OpenAIChat:
-        p = openai_chat_params
-        _tool_choice: Union[str, ChatCompletionNamedToolChoiceParam] = (
-            ModelDescriptorMapper._from_tool_choice(
-                openai_chat_params=openai_chat_params
-            )
-        )
-        _tools: List[ChatCompletionToolParam] = []
-
-        if p.HasField("tools_spec"):
-            for tool in p.tools_spec.tools:
-                _tools.append(json_format.MessageToDict(tool))
-
-        return OpenAIChat(
-            model_id=model_id,
-            frequency_penalty=(
-                p.frequency_penalty if p.HasField("frequency_penalty") else None
-            ),
-            logit_bias=(
-                {k: int(v) for k, v in p.logit_bias.items()}
-                if p.HasField("logit_bias")
-                else None
-            ),
-            logprobs=p.logprobs if p.HasField("logprobs") else None,
-            max_tokens=p.max_tokens if p.HasField("max_tokens") else None,
-            n=p.n if p.HasField("n") else None,
-            presence_penalty=(
-                p.presence_penalty if p.HasField("presence_penalty") else None
-            ),
-            response_format=(
-                p.response_format if p.HasField("response_format") else None
-            ),  # noqa
-            seed=p.seed if p.HasField("seed") else None,
-            stop=[_ for _ in p.stop] if p.HasField("stop") else None,
-            temperature=p.temperature if p.HasField("temperature") else None,
-            top_p=p.top_p if p.HasField("top_p") else None,
-            top_logprobs=p.top_logprobs if p.HasField("top_logprobs") else None,
-            user=p.user if p.HasField("user") else None,
-            tool_choice=_tool_choice if p.HasField("tool_choice") else None,
-            tools=_tools if p.HasField("tools_spec") else None,
-        )
-
-    @staticmethod
-    def from_prompt_openai_provider(
-        open_ai_provider: ProtoPromptOpenAIProvider,
-    ) -> ModelDescriptor:
-        descriptor: ProtoOpenAIModelDescriptor = (
-            open_ai_provider.open_ai_model_descriptor
-        )
-        model_id: str = descriptor.model_id
-
-        if descriptor.capabilities.WhichOneof("optional_chat"):
-            return ModelDescriptorMapper.from_openai_chat_capability(
-                model_id=model_id,
-                openai_chat_params=descriptor.capabilities.chat_api.chat_params,
-            )
-        else:
-            raise FrogmlException(
-                f"Got unsupported openai capability: {repr(open_ai_provider)}"
-            )
-
-    @staticmethod
-    def from_prompt_model_descriptor(
-        model_descriptor: ProtoPromptModelDescriptor,
-    ) -> ModelDescriptor:
-        if model_descriptor.WhichOneof("model_provider") == "open_ai_provider":
-            return ModelDescriptorMapper.from_prompt_openai_provider(
-                model_descriptor.open_ai_provider
-            )
-        else:
-            raise FrogmlException(
-                f"Got unsupported model descriptor: {repr(model_descriptor)}"
-            )
-
-    @staticmethod
-    def to_openai_chat(model_descriptor: OpenAIChat) -> ProtoOpenAIModelDescriptor:
-        d: OpenAIChat = model_descriptor
-        logit_bias_struct = Struct()
-
-        logit_bias_struct.update(d.logit_bias) if d.logit_bias else None
-
-        stop_list_value = ListValue()
-        stop_list_value.extend(d.stop) if d.stop else None
-        tool_choice_proto: ProtoToolChoice
-
-        if isinstance(d.tool_choice, str):
-            tool_choice_proto = ProtoToolChoice(literal=d.tool_choice)
-        elif d.tool_choice is not None:
-            tool_choice_struct = Struct()
-            tool_choice_struct.update(d.tool_choice)
-            tool_choice_proto = ProtoToolChoice(json=tool_choice_struct)
-
-        tools_structs = []
-        if d.tools:
-            for tool in d.tools:
-                s = Struct()
-                s.update(tool)
-                tools_structs.append(s)
-
-        tools_proto = ProtoTools(tools=tools_structs)
-
-        model_capabilities = ProtoOpenAICapabilities(
-            chat_api=ProtoOpenAIChatAPI(
-                chat_params=ProtoOpenAIChatModelParams(
-                    frequency_penalty=d.frequency_penalty,
-                    logit_bias=logit_bias_struct if d.logit_bias else None,
-                    logprobs=d.logprobs,
-                    max_tokens=d.max_tokens,
-                    n=d.n,
-                    presence_penalty=d.presence_penalty,
-                    response_format=(
-                        d.response_format if d.response_format else None
-                    ),  # noqa
-                    seed=d.seed,
-                    stop=stop_list_value if d.stop else None,
-                    temperature=d.temperature,
-                    tool_choice=tool_choice_proto if d.tool_choice else None,
-                    tools_spec=tools_proto if d.tools else None,
-                    top_logprobs=d.top_logprobs,
-                    top_p=d.top_p,
-                    user=d.user,
-                )
-            )
-        )
-
-        return ProtoOpenAIModelDescriptor(
-            model_id=model_descriptor.model_id, capabilities=model_capabilities
-        )
-
-    @staticmethod
-    def to_model_descriptor(
-        model_descriptor: ModelDescriptor,
-    ) -> Union[ProtoOpenAIModelDescriptor]:
-        if isinstance(model_descriptor, OpenAIChat):
-            return ModelDescriptorMapper.to_openai_chat(model_descriptor)
-
-        raise FrogmlException(
-            f"Got unsupported model descriptor: {repr(model_descriptor)}"
-        )

frogml_core-0.0.113/frogml_core/clients/prompt_manager/prompt_manager_client.py

@@ -1,190 +0,0 @@
-import grpc
-from dependency_injector.wiring import Provide, inject
-
-from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
-    CreatePromptRequest as ProtoCreatePromptRequest,
-)
-from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
-    CreatePromptResponse as ProtoCreatePromptResponse,
-)
-from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
-    CreatePromptVersionRequest as ProtoCreatePromptVersionRequest,
-)
-from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
-    CreatePromptVersionResponse as ProtoCreatePromptVersionResponse,
-)
-from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
-    DeletePromptRequest as ProtoDeletePromptRequest,
-)
-from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
-    DeletePromptVersionRequest as ProtoDeletePromptVersionRequest,
-)
-from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
-    GetPromptByNameRequest as ProtoGetPromptByNameRequest,
-)
-from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
-    GetPromptByNameResponse as ProtoGetPromptByNameResponse,
-)
-from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
-    GetPromptVersionByPromptNameRequest as ProtoGetPromptVersionByPromptNameRequest,
-)
-from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
-    GetPromptVersionByPromptNameResponse as ProtoGetPromptVersionByPromptNameResponse,
-)
-from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2 import (
-    SetDefaultPromptVersionRequest as ProtoSetDefaultPromptVersionRequest,
-)
-from frogml_proto.qwak.prompt.v1.prompt.prompt_manager_service_pb2_grpc import (
-    PromptManagerServiceStub,
-)
-from frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import Prompt as ProtoPrompt
-from frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import PromptSpec as ProtoPromptSpec
-from frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import (
-    PromptVersion as ProtoPromptVersion,
-)
-from frogml_proto.qwak.prompt.v1.prompt.prompt_pb2 import (
-    PromptVersionSpec as ProtoPromptVersionSpec,
-)
-from frogml_core.exceptions import FrogmlException
-from frogml_core.inner.di_configuration import FrogmlContainer
-
-
-class PromptManagerClient:
-    @inject
-    def __init__(self, grpc_channel=Provide[FrogmlContainer.core_grpc_channel]):
-        self._grpc_client: PromptManagerServiceStub = PromptManagerServiceStub(
-            grpc_channel
-        )
-
-    def create_prompt(
-        self,
-        name: str,
-        prompt_description: str,
-        version_spec: ProtoPromptVersionSpec,
-    ) -> ProtoPrompt:
-        request = ProtoCreatePromptRequest(
-            prompt_name=name,
-            prompt_spec=ProtoPromptSpec(description=prompt_description),
-            prompt_version_spec=version_spec,
-        )
-        try:
-            response: ProtoCreatePromptResponse = self._grpc_client.CreatePrompt(
-                request
-            )
-            return response.prompt
-        except grpc.RpcError as error:
-            call: grpc.Call = error  # noqa
-            if call.code() == grpc.StatusCode.ALREADY_EXISTS:
-                raise FrogmlException(f"Prompt with name: {name} already exists")
-            elif call.code() == grpc.StatusCode.INVALID_ARGUMENT:
-                raise FrogmlException(
-                    f"Got an illegal prompt specification: {call.details()}"
-                )
-            else:
-                raise FrogmlException(f"Internal Error: {call.details()}")
-
-    def create_prompt_version(
-        self,
-        name: str,
-        version_spec: ProtoPromptVersionSpec,
-        set_default: bool,
-    ) -> ProtoPromptVersion:
-        request = ProtoCreatePromptVersionRequest(
-            prompt_name=name, prompt_version_spec=version_spec, set_default=set_default
-        )
-        try:
-            response: ProtoCreatePromptVersionResponse = (
-                self._grpc_client.CreatePromptVersion(request)
-            )
-            return response.prompt_version
-        except grpc.RpcError as error:
-            call: grpc.Call = error  # noqa
-            if call.code() == grpc.StatusCode.NOT_FOUND:
-                raise FrogmlException(
-                    f"Can not update prompt: '{name}', prompt was not found"
-                )
-            elif call.code() == grpc.StatusCode.INVALID_ARGUMENT:
-                raise FrogmlException(
-                    f"Got an illegal prompt specification: {call.details()}"
-                )
-            else:
-                raise FrogmlException(f"Internal Error: {call.details()}")
-
-    def delete_prompt(self, name: str):
-        try:
-            self._grpc_client.DeletePrompt(ProtoDeletePromptRequest(prompt_name=name))
-        except grpc.RpcError as error:
-            call: grpc.Call = error  # noqa
-            if call.code() == grpc.StatusCode.NOT_FOUND:
-                raise FrogmlException(f"Prompt named '{name}' was not found")
-            else:
-                raise FrogmlException(f"Internal Error: {call.details()}")
-
-    def delete_prompt_version(self, name: str, version: int):
-        try:
-            self._grpc_client.DeletePromptVersion(
-                ProtoDeletePromptVersionRequest(
-                    prompt_name=name, version_number=version
-                )
-            )
-        except grpc.RpcError as error:
-            call: grpc.Call = error  # noqa
-            if call.code() == grpc.StatusCode.NOT_FOUND:
-                raise FrogmlException(str(call.details()))
-            elif call.code() == grpc.StatusCode.FAILED_PRECONDITION:
-                raise FrogmlException(
-                    f"Cannot delete the default version '{version}' of a prompt '{name}',"
-                    f" please set another version as the default to delete this version."
-                )
-            else:
-                raise FrogmlException(f"Internal Error: {call.details()}")
-
-    def get_prompt_by_name(self, name: str) -> ProtoPrompt:
-        """
-        Get prompt's default version
-        """
-        try:
-            response: ProtoGetPromptByNameResponse = self._grpc_client.GetPromptByName(
-                ProtoGetPromptByNameRequest(prompt_name=name)
-            )
-            return response.prompt
-        except grpc.RpcError as error:
-            call: grpc.Call = error  # noqa
-            if call.code() == grpc.StatusCode.NOT_FOUND:
-                raise FrogmlException(str(call.details()))
-            else:
-                raise FrogmlException(f"Internal Error: {call.details()}")
-
-    def get_prompt_version_by_name(self, name: str, version: int) -> ProtoPromptVersion:
-        """
-        Get prompt specific version
-        """
-        try:
-            response: ProtoGetPromptVersionByPromptNameResponse = (
-                self._grpc_client.GetPromptVersionByPromptName(
-                    ProtoGetPromptVersionByPromptNameRequest(
-                        prompt_name=name, version_number=version
-                    )
-                )
-            )
-            return response.prompt_version
-        except grpc.RpcError as error:
-            call: grpc.Call = error  # noqa
-            if call.code() == grpc.StatusCode.NOT_FOUND:
-                raise FrogmlException(str(call.details()))
-            else:
-                raise FrogmlException(f"Internal Error: {call.details()}")
-
-    def set_default_prompt_version(self, name: str, version: int):
-        try:
-            self._grpc_client.SetDefaultPromptVersion(
-                ProtoSetDefaultPromptVersionRequest(
-                    prompt_name=name, version_number=version
-                )
-            )
-        except grpc.RpcError as error:
-            call: grpc.Call = error  # noqa
-            if call.code() == grpc.StatusCode.NOT_FOUND:
-                raise FrogmlException(str(call.details()))
-            else:
-                raise FrogmlException(f"Internal Error: {call.details()}")