This commit is contained in:
Changhua
2023-12-15 01:54:15 +08:00
parent 1fb34e9e04
commit b8bb65ec88

View File

@@ -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