Skip to content

Commit

Permalink
modify type hints in protected methods
Browse files Browse the repository at this point in the history
  • Loading branch information
AlistairLR112 committed Jan 3, 2024
1 parent e466bd3 commit 1cc5ce8
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions integrations/ollama/src/ollama_haystack/generator.py
Original file line number Diff line number Diff line change
Expand Up @@ -50,13 +50,13 @@ def __init__(
self.url = url
self.generation_kwargs = generation_kwargs or {}

def _get_telemetry_data(self) -> Dict[str, Any]:
def _get_telemetry_data(self) -> Dict[str, str]:
"""
Data that is sent to Posthog for usage analytics.
"""
return {"model": self.model_name}

def _post_args(self, prompt: str, generation_kwargs=None) -> Dict[str, Any]:
def _post_args(self, prompt: str, generation_kwargs=None) -> Dict[str, Union[str, dict]]:
"""
Returns A dictionary of arguments for a POST request to an Ollama service
:param prompt: the prompt to generate a response for
Expand Down

0 comments on commit 1cc5ce8

Please sign in to comment.