From 4f5c4d1e3b8fdd30e5f1a7445a765a27e230e632 Mon Sep 17 00:00:00 2001
From: Dinesh Yeduguru <dineshyv@fb.com>
Date: Thu, 7 Nov 2024 21:40:54 -0800
Subject: [PATCH] add back llama_model field

---
 llama_stack/apis/models/models.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/llama_stack/apis/models/models.py b/llama_stack/apis/models/models.py
index d11f31ab6..d50d5a656 100644
--- a/llama_stack/apis/models/models.py
+++ b/llama_stack/apis/models/models.py
@@ -15,6 +15,9 @@ from llama_stack.apis.resource import Resource
 @json_schema_type
 class Model(Resource):
     type: Literal["model"] = "model"
+    llama_model: str = Field(
+        description="Pointer to the underlying core Llama family model. Each model served by Llama Stack must have a core Llama model.",
+    )
     metadata: Dict[str, Any] = Field(
         default_factory=dict,
         description="Any additional metadata for this model",