From b8bb65ec88d2e7fece231c5fec39dfe300293402 Mon Sep 17 00:00:00 2001 From: Changhua Date: Fri, 15 Dec 2023 01:54:15 +0800 Subject: [PATCH] Reformat --- base/func_bard.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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