Add HTTP models for BrainForge service endpoints

neon_data_models/models/api/http/brainforge.py

@@ -1,9 +1,25 @@
-from typing import List
+from typing import List, Optional
 from pydantic import Field
 
 from neon_data_models.models.base import BaseModel
-from neon_data_models.models.api.llm import BrainForgeLLM
+from neon_data_models.models.api.llm import BrainForgeLLM, LLMPersona, LLMRequest
 
 
 class LLMGetModelsHttpResponse(BaseModel):
     models: List[BrainForgeLLM]
+
+
+class LLMGetPersonasHttpRequest(BaseModel):
+    model_id: str = Field(
+        description="Model ID (<name>@<version>) to get personas for")
+
+
+class LLMGetPersonasHttpResponse(BaseModel):
+    personas: List[LLMPersona] = Field(
+        description="List of personas associated with the requested model.")
+
+
+class LLMGetInferenceHttpRequest(LLMRequest):
+    llm_name: str = Field(description="Model name to request")
+    llm_revision: str = Field(description="Model revision to request")
+    model: Optional[str] = None

neon_data_models/models/api/mq/brainforge.py

@@ -4,7 +4,7 @@
 
 from neon_data_models.models.base.contexts import MQContext
 from neon_data_models.models.api.llm import BrainForgeLLM, LLMRequest, LLMResponse, LLMPersona
-from neon_data_models.models.api.http.brainforge import LLMGetModelsHttpResponse
+from neon_data_models.models.api.http.brainforge import LLMGetModelsHttpResponse, LLMGetPersonasHttpRequest
 
 
 class LLMGetModels(MQContext):
@@ -16,10 +16,7 @@ class LLMGetModelsResponse(MQContext, LLMGetModelsHttpResponse):
     pass
 
 
-class LLMGetPersonas(LLMGetModels):
-    model_id: str = Field(
-        description="Model ID (<name>@<version>) to get personas for")
-
+class LLMGetPersonas(LLMGetModels, LLMGetPersonasHttpRequest):
     @property
     def model_name(self):
         return self.model_id.split("@")[0]