fix ai_bard bug

This commit is contained in:
Chill
2024-11-06 17:09:03 +08:00
parent 3acdfc324c
commit 49dbf7b1b8
3 changed files with 7 additions and 7 deletions

View File

@@ -8,7 +8,7 @@
{ {
'name': 'Google Bard Ai for odoo ai center, 谷歌Ai支持', 'name': 'Google Bard Ai for odoo ai center, 谷歌Ai支持',
'version': '16.23.10.11', 'version': '16.24.11.06',
'author': 'odooai.cn', 'author': 'odooai.cn',
'company': 'odooai.cn', 'company': 'odooai.cn',
'maintainer': 'odooai.cn', 'maintainer': 'odooai.cn',

View File

@@ -18,7 +18,7 @@ class AiRobot(models.Model):
provider = fields.Selection( provider = fields.Selection(
selection_add=[('google', 'Google Ai')], selection_add=[('google', 'Google Ai')],
ondelete={'google': 'set default'} ondelete={'google': 'set default'}
), )
set_ai_model = fields.Selection( set_ai_model = fields.Selection(
selection_add=[('google-bard', 'Google Bard')], selection_add=[('google-bard', 'Google Bard')],
ondelete={'google-bard': 'set default'}) ondelete={'google-bard': 'set default'})
@@ -28,13 +28,13 @@ class AiRobot(models.Model):
if self.provider == 'google': if self.provider == 'google':
self.endpoint = 'https://api.bard.ai/v1/text/generate' self.endpoint = 'https://api.bard.ai/v1/text/generate'
return super()._onchange_provider() return super()._onchange_provider()
def get_google(self, data, author_id, answer_id, param={}): def get_google(self, data, author_id, answer_id, param={}):
self.ensure_one() self.ensure_one()
headers = {"Content-Type": "application/json", "Authorization": f"Bearer {self.openapi_api_key}"} headers = {"Content-Type": "application/json", "Authorization": f"Bearer {self.openapi_api_key}"}
R_TIMEOUT = self.ai_timeout or 120 R_TIMEOUT = self.ai_timeout or 120
o_url = self.endpoint or "https://api.bard.ai/v1/text/generate" o_url = self.endpoint or "https://api.bard.ai/v1/text/generate"
# todo: 更多参数如 prompt, max_length # todo: 更多参数如 prompt, max_length
max_tokens = param.get('max_tokens') if param.get('max_tokens') else self.max_tokens max_tokens = param.get('max_tokens') if param.get('max_tokens') else self.max_tokens
temperature = param.get('temperature') if param.get('temperature') else self.temperature temperature = param.get('temperature') if param.get('temperature') else self.temperature
@@ -52,9 +52,9 @@ class AiRobot(models.Model):
_logger.warning("Get Response Json failed: %s", e) _logger.warning("Get Response Json failed: %s", e)
else: else:
_logger.warning('=====================Openai output data: %s' % response.json()) _logger.warning('=====================Openai output data: %s' % response.json())
def get_google_post(self, res, author_id=False, answer_id=False, param={}): def get_google_post(self, res, author_id=False, answer_id=False, param={}):
if self.provider == 'google': if self.provider == 'google':
content = res['text'] content = res['text']
return content, False, True return content, False, True

View File

@@ -4,7 +4,7 @@
<h2 class="oe_slogan"> Google Bard Ai for odoo ai center</h2> <h2 class="oe_slogan"> Google Bard Ai for odoo ai center</h2>
<h3 class="oe_slogan"> Ai center addons. all aigc in one. </h3> <h3 class="oe_slogan"> Ai center addons. all aigc in one. </h3>
<div class="oe_row"> <div class="oe_row">
<h3>Latest update: v16.23.10.11</h3> <h3>Latest update: v16.24.11.06</h3>
<div class="row"> <div class="row">
<div class="row"> <div class="row">
<img class="" src="bard.gif"> <img class="" src="bard.gif">