diff --git a/base/func_bard.py b/base/func_bard.py index b7fb8d3..a2bfa6d 100644 --- a/base/func_bard.py +++ b/base/func_bard.py @@ -1,7 +1,9 @@ #! /usr/bin/env python3 # -*- coding: utf-8 -*- + import google.generativeai as genai + class BardAssistant: def __init__(self, conf: dict) -> None: self._api_key = conf["api_key"] @@ -9,7 +11,6 @@ class BardAssistant: genai.configure(api_key=self._api_key) self._bard = genai.GenerativeModel(self._model_name) - def __repr__(self): return 'BardAssistant' @@ -23,6 +24,7 @@ class BardAssistant: response = self._bard.generate_content(msg) return response.text + if __name__ == "__main__": from configuration import Config config = Config().BardAssistant