Merge pull request #12 from Serverless-Devs/copilot/sub-pr-11

[WIP] WIP Address feedback on Google ADK support

Author: OhYee

agentrun/model/api/data.py

@@ -54,7 +54,7 @@ def completions(
             **self.headers,
             **kwargs.get("headers", {}),
         }
-        if kwargs["stream_options"] is None:
+        if kwargs.get("stream_options") is None:
             kwargs["stream_options"] = {}
         kwargs["stream_options"]["include_usage"] = True
 
@@ -86,7 +86,7 @@ def responses(
             **self.headers,
             **kwargs.get("headers", {}),
         }
-        if kwargs["stream_options"] is None:
+        if kwargs.get("stream_options") is None:
             kwargs["stream_options"] = {}
         kwargs["stream_options"]["include_usage"] = True
         from litellm import responses