|
14 | 14 | os.environ["VM_API_MODEL"] = "your_model" |
15 | 15 |
|
16 | 16 | import validmind.api_client as api_client |
| 17 | +from validmind.__version__ import __version__ |
17 | 18 | from validmind.errors import ( |
18 | 19 | MissingAPICredentialsError, |
19 | 20 | MissingModelIdError, |
@@ -87,6 +88,7 @@ def test_init_successful(self, mock_requests_get): |
87 | 88 | "X-API-SECRET": os.environ["VM_API_SECRET"], |
88 | 89 | "X-MODEL-CUID": os.environ["VM_API_MODEL"], |
89 | 90 | "X-MONITORING": "False", |
| 91 | + "X-LIBRARY-VERSION": __version__, |
90 | 92 | }, |
91 | 93 | ) |
92 | 94 |
|
@@ -129,6 +131,7 @@ def test_init_no_warning_when_document_is_passed( |
129 | 131 | "X-API-SECRET": os.environ["VM_API_SECRET"], |
130 | 132 | "X-MODEL-CUID": os.environ["VM_API_MODEL"], |
131 | 133 | "X-MONITORING": "False", |
| 134 | + "X-LIBRARY-VERSION": __version__, |
132 | 135 | "X-DOCUMENT-TYPE": "documentation", |
133 | 136 | }, |
134 | 137 | ) |
@@ -184,6 +187,7 @@ def test_init_unsuccessful_ping(self, mock_get): |
184 | 187 | "X-API-SECRET": os.environ["VM_API_SECRET"], |
185 | 188 | "X-MODEL-CUID": os.environ["VM_API_MODEL"], |
186 | 189 | "X-MONITORING": "False", |
| 190 | + "X-LIBRARY-VERSION": __version__, |
187 | 191 | }, |
188 | 192 | ) |
189 | 193 |
|
|
0 commit comments