liuwei_x_x/abot
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Reformat

Browse Source
This commit is contained in:
Changhua
2023-12-15 01:54:15 +08:00
parent 1fb34e9e04
commit b8bb65ec88
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
base/func_bard.py
View File

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