Skip to content

Commit e52faaf

Browse files
committed
mypy part 2
1 parent eecf632 commit e52faaf

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/neo4j_graphrag/llm/vertexai_llm.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -255,7 +255,7 @@ async def _acall_llm(
255255
model = self._get_model(system_instruction=system_instruction)
256256
options = self._get_call_params(input, message_history, tools)
257257
response = await model.generate_content_async(**options)
258-
return response
258+
return response # type: ignore[no-any-return]
259259

260260
def _call_llm(
261261
self,
@@ -267,7 +267,7 @@ def _call_llm(
267267
model = self._get_model(system_instruction=system_instruction)
268268
options = self._get_call_params(input, message_history, tools)
269269
response = model.generate_content(**options)
270-
return response
270+
return response # type: ignore[no-any-return]
271271

272272
def _to_tool_call(self, function_call: FunctionCall) -> ToolCall:
273273
return ToolCall(

0 commit comments

Comments
 (0)