Add schema for passing full workspace config to CREATE workspace API (#1086)

This adds an optional full workspace definition so we can take exported
workspaces into use.

Signed-off-by: Juan Antonio Osorio <[email protected]>

Author: JAORMX

src/codegate/api/v1_models.py

@@ -5,6 +5,7 @@
 
 import pydantic
 
+import codegate.muxing.models as mux_models
 from codegate.db import models as db_models
 from codegate.extract_snippets.message_extractor import CodeSnippet
 from codegate.providers.base import BaseProvider
@@ -59,9 +60,19 @@ def from_db_workspaces(
         )
 
 
-class CreateOrRenameWorkspaceRequest(pydantic.BaseModel):
+class WorkspaceConfig(pydantic.BaseModel):
+    system_prompt: str
+
+    muxing_rules: List[mux_models.MuxRule]
+
+
+class FullWorkspace(pydantic.BaseModel):
     name: str
 
+    config: Optional[WorkspaceConfig] = None
+
+
+class CreateOrRenameWorkspaceRequest(FullWorkspace):
     # If set, rename the workspace to this name. Note that
     # the 'name' field is still required and the workspace
     # workspace must exist.

src/codegate/muxing/models.py

@@ -26,6 +26,8 @@ class MuxRule(pydantic.BaseModel):
     Represents a mux rule for a provider.
     """
 
+    # Used for exportable workspaces
+    provider_name: Optional[str] = None
     provider_id: str
     model: str
     # The type of matcher to use