updates

Signed-off-by: Jess Frazelle <github@jessfraz.com>

kittycad/api/ai/__init__.py

@@ -1 +0,0 @@
-""" Contains methods for accessing the ai API paths: AI uses machine learning to generate CAD models. """  # noqa: E501

kittycad/api/ml/__init__.py

@@ -0,0 +1 @@
+""" Contains methods for accessing the ml API paths: Machine learning to generate CAD models and other things. """  # noqa: E501

kittycad/client_test.py

@@ -6,7 +6,6 @@ from typing import Optional, Union
 
 import pytest
 
-from .api.ai import create_text_to_cad, get_text_to_cad_model_for_user
 from .api.api_tokens import list_api_tokens_for_user
 from .api.file import (
     create_file_center_of_mass,
@@ -15,6 +14,7 @@ from .api.file import (
     create_file_volume,
 )
 from .api.meta import ping
+from .api.ml import create_text_to_cad, get_text_to_cad_model_for_user
 from .api.modeling import modeling_commands_ws
 from .api.users import get_user_self, list_users_extended
 from .client import ClientFromEnv

kittycad/examples_test.py

@@ -2,15 +2,6 @@ from typing import List, Optional, Union
 
 import pytest
 
-from kittycad.api.ai import (
-    create_kcl_code_completions,
-    create_text_to_cad,
-    create_text_to_cad_model_feedback,
-    get_ai_prompt,
-    get_text_to_cad_model_for_user,
-    list_ai_prompts,
-    list_text_to_cad_models_for_user,
-)
 from kittycad.api.api_calls import (
     get_api_call,
     get_api_call_for_org,
@@ -60,6 +51,15 @@ from kittycad.api.meta import (
     internal_get_api_token_for_discord_user,
     ping,
 )
+from kittycad.api.ml import (
+    create_kcl_code_completions,
+    create_text_to_cad,
+    create_text_to_cad_model_feedback,
+    get_ai_prompt,
+    get_text_to_cad_model_for_user,
+    list_ai_prompts,
+    list_text_to_cad_models_for_user,
+)
 from kittycad.api.modeling import modeling_commands_ws
 from kittycad.api.orgs import (
     create_org,

kittycad/models/__init__.py

@@ -5,6 +5,7 @@ from .add_org_member import AddOrgMember
 from .address_details import AddressDetails
 from .ai_feedback import AiFeedback
 from .ai_prompt import AiPrompt
+from .ai_prompt_metadata import AiPromptMetadata
 from .ai_prompt_results_page import AiPromptResultsPage
 from .ai_prompt_type import AiPromptType
 from .angle import Angle

kittycad/models/ai_prompt.py

@@ -1,9 +1,10 @@
 import datetime
-from typing import Any, Optional
+from typing import Optional
 
 from pydantic import BaseModel, ConfigDict
 
 from ..models.ai_feedback import AiFeedback
+from ..models.ai_prompt_metadata import AiPromptMetadata
 from ..models.ai_prompt_type import AiPromptType
 from ..models.api_call_status import ApiCallStatus
 from ..models.uuid import Uuid
@@ -22,7 +23,7 @@ class AiPrompt(BaseModel):
 
     id: Uuid
 
-    metadata: Optional[Any] = None
+    metadata: Optional[AiPromptMetadata] = None
 
     model_version: str
 

kittycad/models/ai_prompt_metadata.py

@@ -0,0 +1,12 @@
+from typing import Optional
+
+from pydantic import BaseModel, ConfigDict
+
+
+
+class AiPromptMetadata(BaseModel):
+    """Metadata for an AI prompt."""
+
+    code: Optional[str] = None
+
+    model_config = ConfigDict(protected_namespaces=())

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "kittycad"
-version = "0.6.18"
+version = "0.6.19"
 description = "A client library for accessing KittyCAD"
 
 authors = []