dek924 commited on
Commit
2863a63
·
1 Parent(s): 11786ab

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
  )