Add x-ms-original-file in generated sample to declare original samp…

…le file of typespec (#2805) * original file in sample * changelog * fix for ci
Azure · Sep 2, 2024 · 8ad6754 · 8ad6754
1 parent 3856eb4
commit 8ad6754
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 1 deletion.
diff --git a/.chronus/changes/sample-dir-2024-8-2-13-40-48.md b/.chronus/changes/sample-dir-2024-8-2-13-40-48.md
@@ -0,0 +1,7 @@
+---
+changeKind: feature
+packages:
+  - "@azure-tools/typespec-python"
+---
+
+Add `x-ms-original-file` in generated sample to declare original sample file of typespec
diff --git a/packages/typespec-python/generator/pygen/codegen/serializers/sample_serializer.py b/packages/typespec-python/generator/pygen/codegen/serializers/sample_serializer.py
@@ -150,7 +150,7 @@ def _origin_file(self) -> str:
         name = self.sample.get("x-ms-original-file", "")
         if "specification" in name:
             return "specification" + name.split("specification")[-1]
-        return ""
+        return name if self.code_model.options["from_typespec"] else ""
 
     def serialize(self) -> str:
         operation_result, return_var = self._operation_result()

diff --git a/packages/typespec-python/test/azure/generated/azure-example-basic/generated_samples/basic.py b/packages/typespec-python/test/azure/generated/azure-example-basic/generated_samples/basic.py
@@ -32,5 +32,6 @@ def main():
     print(response)
 
 
+# x-ms-original-file: 2022-12-01-preview/basic.json
 if __name__ == "__main__":
     main()