Reformat
This commit is contained in:
@@ -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
|
||||
|
||||
Reference in New Issue
Block a user