We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 98bb315 commit aca79a4Copy full SHA for aca79a4
1 file changed
integrations/ollama/tests/test_chat_generator.py
@@ -137,7 +137,7 @@ def test_convert_ollama_response_to_chatmessage(self):
137
assert observed.role == "assistant"
138
assert observed.text == "Hello! How are you today?"
139
140
- assert observed.meta == {
+ expected_data = {
141
"finish_reason": "stop",
142
"usage": {
143
"completion_tokens": 298,
@@ -153,6 +153,11 @@ def test_convert_ollama_response_to_chatmessage(self):
153
"model": "some_model",
154
}
155
156
+ if "logprobs" in observed.meta:
157
+ expected_data["logprobs"] = None
158
+
159
+ assert observed.meta == expected_data
160
161
def test_convert_ollama_response_to_chatmessage_with_tools(self):
162
model = "some_model"
163
0 commit comments