Skip to content

Commit e70550d

Browse files
andrewheardandrewheard
committed
Update test JSON payloads with Vertex AI output (#12615)
1 parent af52205 commit e70550d

File tree

6 files changed

+75
-86
lines changed

6 files changed

+75
-86
lines changed

‎FirebaseVertexAI/Tests/Unit/ChatTests.swift

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -66,7 +66,7 @@ final class ChatTests: XCTestCase {
6666
XCTAssertEqual(chat.history.count, 2)
6767
XCTAssertEqual(chat.history[0].parts[0].text, input)
6868

69-
let finalText = "1 2 3 4 5 6 7 8 9 10"
69+
let finalText = "1 2 3 4 5 6 7 8"
7070
let assembledExpectation = ModelContent(role: "model", parts: finalText)
7171
XCTAssertEqual(chat.history[0].parts[0].text, input)
7272
XCTAssertEqual(chat.history[1], assembledExpectation)

‎FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-long.txt

Lines changed: 8 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -1,19 +1,15 @@
1-
data: {"candidates": [{"content": {"parts": [{"text": "1 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
1+
data: {"candidates": [{"content": {"role": "model","parts": [{"text": "1 "}]}}]}
22

3-
data: {"candidates": [{"content": {"parts": [{"text": "2 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
3+
data: {"candidates": [{"content": {"role": "model","parts": [{"text": "2 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.0394904,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04468087},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.034553625,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03890198},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.09401018,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.025809621},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.036562506,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.047691282}]}]}
44

5-
data: {"candidates": [{"content": {"parts": [{"text": "3 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
5+
data: {"candidates": [{"content": {"role": "model","parts": [{"text": "3 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.03507868,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.045183755},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.027742893,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.043528143},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.08803312,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.026105914},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.06681233,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.053899158}]}]}
66

7-
data: {"candidates": [{"content": {"parts": [{"text": "4 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
7+
data: {"candidates": [{"content": {"role": "model","parts": [{"text": "4 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.037750278,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05089372},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.040087357,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05888469},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.071202725,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.034100424},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.07613248,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.051749535}]}]}
88

9-
data: {"candidates": [{"content": {"parts": [{"text": "5 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
9+
data: {"candidates": [{"content": {"role": "model","parts": [{"text": "5 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.04672496,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.059210256},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.04977345,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05623635},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.083890386,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.03825006},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.08359067,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05975658}]}]}
1010

11-
data: {"candidates": [{"content": {"parts": [{"text": "6 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
11+
data: {"candidates": [{"content": {"role": "model","parts": [{"text": "6 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.07779744,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.06052939},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.041930523,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.056756895},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.12787028,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05350215},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.09203286,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.048676573}]}]}
1212

13-
data: {"candidates": [{"content": {"parts": [{"text": "7 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
13+
data: {"candidates": [{"content": {"role": "model","parts": [{"text": "7 "}]},"safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.071202725,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.05291181},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.031439852,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04509957},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.11417085,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04922211},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.09451043,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.052716404}]}]}
1414

15-
data: {"candidates": [{"content": {"parts": [{"text": "8 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
16-
17-
data: {"candidates": [{"content": {"parts": [{"text": "9 "}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
18-
19-
data: {"candidates": [{"content": {"parts": [{"text": "10"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}]}
15+
data: {"candidates": [{"content": {"role": "model","parts": [{"text": "8"}]},"finishReason": "STOP","safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.06221698,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.045777276},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.03085051,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.04560694},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.0992954,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.040769264},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.100701615,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.061424047}]}],"usageMetadata": {"promptTokenCount": 6,"candidatesTokenCount": 326,"totalTokenCount": 332}}

‎FirebaseVertexAI/Tests/Unit/GenerateContentResponses/streaming-success-basic-reply-short.txt

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1 @@
1-
data: {"candidates": [{"content": {"parts": [{"text": "Cheyenne"}]},"finishReason": "STOP","index": 0,"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}],"promptFeedback": {"safetyRatings": [{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE"},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE"}]}}
2-
1+
data: {"candidates": [{"content": {"role": "model","parts": [{"text": "Mountain View, California"}]},"finishReason": "STOP","safetyRatings": [{"category": "HARM_CATEGORY_HATE_SPEECH","probability": "NEGLIGIBLE","probabilityScore": 0.02854415,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.052424565},{"category": "HARM_CATEGORY_DANGEROUS_CONTENT","probability": "NEGLIGIBLE","probabilityScore": 0.24926445,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.0996453},{"category": "HARM_CATEGORY_HARASSMENT","probability": "NEGLIGIBLE","probabilityScore": 0.087096825,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.043123372},{"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT","probability": "NEGLIGIBLE","probabilityScore": 0.14402841,"severity": "HARM_SEVERITY_NEGLIGIBLE","severityScore": 0.086169556}]}],"usageMetadata": {"promptTokenCount": 6,"candidatesTokenCount": 4,"totalTokenCount": 10}}

‎FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-long.json

Lines changed: 26 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -2,53 +2,49 @@
22
"candidates": [
33
{
44
"content": {
5+
"role": "model",
56
"parts": [
67
{
7-
"text": "You can ask me a wide range of questions on various topics. Here are some examples:\n\n1. **General Knowledge:**\n - What is the capital of France?\n - Who painted the Mona Lisa?\n - What is the largest ocean in the world?\n\n2. **Science and Technology:**\n - How does a computer work?\n - What is the difference between a virus and a bacteria?\n - What are the latest advancements in artificial intelligence?\n\n3. **History and Culture:**\n - Who was the first president of the United States?\n - What is the significance of the Great Wall of China?\n - What are some of the most famous works of Shakespeare?\n\n4. **Current Events:**\n - What is the latest news about the COVID-19 pandemic?\n - Who is the current president of Ukraine?\n - What are the major issues being discussed in the upcoming election?\n\n5. **Personal Questions:**\n - What are your hobbies?\n - What is your favorite book or movie?\n - What are your thoughts on the future of technology?\n\n6. **Fun and Games:**\n - Can you tell me a joke?\n - What is the answer to this riddle: \"I have keys but no locks. I have space but no room. You can enter, but can't go outside.\" (Answer: a keyboard)\n - Let's play a game of 20 questions.\n\n7. **Hypothetical Questions:**\n - What would you do if you won the lottery?\n - What would happen if time travel were possible?\n - What is the meaning of life?\n\n8. **Philosophical Questions:**\n - What is the nature of reality?\n - Does free will exist?\n - What is the difference between right and wrong?\n\n9. **Creative Questions:**\n - Write a poem about a sunset.\n - Design a logo for a new company.\n - Compose a song about your favorite season.\n\n10. **Technical Questions:**\n - How can I improve the performance of my computer?\n - What is the best way to troubleshoot a network issue?\n - How do I create a website using HTML and CSS?\n\nRemember, I am still under development and may not be able to answer all questions perfectly. However, I will do my best to provide you with accurate and informative responses."
8+
"text": "You can ask me a wide range of questions, including:\n\n* **General knowledge questions:** What is the capital of France? Who painted the Mona Lisa?\n* **Science questions:** What is the chemical formula for water? How does photosynthesis work?\n* **History questions:** When did World War II start? Who was the first president of the United States?\n* **Math questions:** What is the square root of 16? How do you solve for x in the equation x + 5 = 10?\n* **Current events questions:** What is happening in Ukraine? Who is the current president of the United States?\n* **Personal questions:** What are your hobbies? What are your favorite books?\n* **Hypothetical questions:** What would happen if all the ice caps melted? What if humans could live on Mars?\n* **Questions about me:** What is your name? How old are you? What are your capabilities?\n\n**Tips for asking questions:**\n\n* Be specific and clear in your questions.\n* Use correct grammar and spelling.\n* Try to ask open-ended questions that allow for multiple answers.\n* Be patient and wait for my response. I am still under development, so I may take a few seconds to process your question.\n\nPlease note that I am still under development and may not be able to answer all questions accurately. However, I will do my best to provide you with the most relevant information I have available."
89
}
9-
],
10-
"role": "model"
10+
]
1111
},
1212
"finishReason": "STOP",
13-
"index": 0,
1413
"safetyRatings": [
1514
{
16-
"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
17-
"probability": "NEGLIGIBLE"
15+
"category": "HARM_CATEGORY_HATE_SPEECH",
16+
"probability": "NEGLIGIBLE",
17+
"probabilityScore": 0.047869004,
18+
"severity": "HARM_SEVERITY_NEGLIGIBLE",
19+
"severityScore": 0.050705366
1820
},
1921
{
20-
"category": "HARM_CATEGORY_HATE_SPEECH",
21-
"probability": "NEGLIGIBLE"
22+
"category": "HARM_CATEGORY_DANGEROUS_CONTENT",
23+
"probability": "NEGLIGIBLE",
24+
"probabilityScore": 0.052134257,
25+
"severity": "HARM_SEVERITY_NEGLIGIBLE",
26+
"severityScore": 0.036288295
2227
},
2328
{
2429
"category": "HARM_CATEGORY_HARASSMENT",
25-
"probability": "NEGLIGIBLE"
30+
"probability": "NEGLIGIBLE",
31+
"probabilityScore": 0.08464396,
32+
"severity": "HARM_SEVERITY_NEGLIGIBLE",
33+
"severityScore": 0.033907957
2634
},
2735
{
28-
"category": "HARM_CATEGORY_DANGEROUS_CONTENT",
29-
"probability": "NEGLIGIBLE"
36+
"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
37+
"probability": "NEGLIGIBLE",
38+
"probabilityScore": 0.06290424,
39+
"severity": "HARM_SEVERITY_NEGLIGIBLE",
40+
"severityScore": 0.050611436
3041
}
3142
]
3243
}
3344
],
34-
"promptFeedback": {
35-
"safetyRatings": [
36-
{
37-
"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
38-
"probability": "NEGLIGIBLE"
39-
},
40-
{
41-
"category": "HARM_CATEGORY_HATE_SPEECH",
42-
"probability": "NEGLIGIBLE"
43-
},
44-
{
45-
"category": "HARM_CATEGORY_HARASSMENT",
46-
"probability": "NEGLIGIBLE"
47-
},
48-
{
49-
"category": "HARM_CATEGORY_DANGEROUS_CONTENT",
50-
"probability": "NEGLIGIBLE"
51-
}
52-
]
45+
"usageMetadata": {
46+
"promptTokenCount": 6,
47+
"candidatesTokenCount": 303,
48+
"totalTokenCount": 309
5349
}
5450
}

‎FirebaseVertexAI/Tests/Unit/GenerateContentResponses/unary-success-basic-reply-short.json

Lines changed: 25 additions & 29 deletions
Original file line numberDiff line numberDiff line change
@@ -2,53 +2,49 @@
22
"candidates": [
33
{
44
"content": {
5+
"role": "model",
56
"parts": [
67
{
78
"text": "Mountain View, California, United States"
89
}
9-
],
10-
"role": "model"
10+
]
1111
},
1212
"finishReason": "STOP",
13-
"index": 0,
1413
"safetyRatings": [
1514
{
16-
"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
17-
"probability": "NEGLIGIBLE"
15+
"category": "HARM_CATEGORY_HATE_SPEECH",
16+
"probability": "NEGLIGIBLE",
17+
"probabilityScore": 0.029035643,
18+
"severity": "HARM_SEVERITY_NEGLIGIBLE",
19+
"severityScore": 0.05613278
1820
},
1921
{
20-
"category": "HARM_CATEGORY_HATE_SPEECH",
21-
"probability": "NEGLIGIBLE"
22+
"category": "HARM_CATEGORY_DANGEROUS_CONTENT",
23+
"probability": "NEGLIGIBLE",
24+
"probabilityScore": 0.2641685,
25+
"severity": "HARM_SEVERITY_NEGLIGIBLE",
26+
"severityScore": 0.082253955
2227
},
2328
{
2429
"category": "HARM_CATEGORY_HARASSMENT",
25-
"probability": "NEGLIGIBLE"
30+
"probability": "NEGLIGIBLE",
31+
"probabilityScore": 0.087252244,
32+
"severity": "HARM_SEVERITY_NEGLIGIBLE",
33+
"severityScore": 0.04509957
2634
},
2735
{
28-
"category": "HARM_CATEGORY_DANGEROUS_CONTENT",
29-
"probability": "NEGLIGIBLE"
36+
"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
37+
"probability": "NEGLIGIBLE",
38+
"probabilityScore": 0.1431877,
39+
"severity": "HARM_SEVERITY_NEGLIGIBLE",
40+
"severityScore": 0.11027937
3041
}
3142
]
3243
}
3344
],
34-
"promptFeedback": {
35-
"safetyRatings": [
36-
{
37-
"category": "HARM_CATEGORY_SEXUALLY_EXPLICIT",
38-
"probability": "NEGLIGIBLE"
39-
},
40-
{
41-
"category": "HARM_CATEGORY_HATE_SPEECH",
42-
"probability": "NEGLIGIBLE"
43-
},
44-
{
45-
"category": "HARM_CATEGORY_HARASSMENT",
46-
"probability": "NEGLIGIBLE"
47-
},
48-
{
49-
"category": "HARM_CATEGORY_DANGEROUS_CONTENT",
50-
"probability": "NEGLIGIBLE"
51-
}
52-
]
45+
"usageMetadata": {
46+
"promptTokenCount": 6,
47+
"candidatesTokenCount": 7,
48+
"totalTokenCount": 13
5349
}
5450
}

0 commit comments

Comments
 (0)