Spaces:
Running
Running
feat: add api meta label
Browse files
patientsim/client/google_client.py
CHANGED
|
@@ -69,6 +69,9 @@ class GeminiClient:
|
|
| 69 |
contents=self.histories,
|
| 70 |
config=types.GenerateContentConfig(
|
| 71 |
system_instruction=system_prompt,
|
|
|
|
|
|
|
|
|
|
| 72 |
**kwargs
|
| 73 |
)
|
| 74 |
)
|
|
|
|
| 69 |
contents=self.histories,
|
| 70 |
config=types.GenerateContentConfig(
|
| 71 |
system_instruction=system_prompt,
|
| 72 |
+
labels={
|
| 73 |
+
"team": "patientsim-demo",
|
| 74 |
+
},
|
| 75 |
**kwargs
|
| 76 |
)
|
| 77 |
)
|
patientsim/client/google_vertex_client.py
CHANGED
|
@@ -73,6 +73,9 @@ class GeminiVertexClient:
|
|
| 73 |
contents=self.histories,
|
| 74 |
config=types.GenerateContentConfig(
|
| 75 |
system_instruction=system_prompt,
|
|
|
|
|
|
|
|
|
|
| 76 |
**kwargs
|
| 77 |
)
|
| 78 |
)
|
|
|
|
| 73 |
contents=self.histories,
|
| 74 |
config=types.GenerateContentConfig(
|
| 75 |
system_instruction=system_prompt,
|
| 76 |
+
labels={
|
| 77 |
+
"team": "patientsim-demo",
|
| 78 |
+
},
|
| 79 |
**kwargs
|
| 80 |
)
|
| 81 |
)
|