mirror of
https://github.com/block/goose.git
synced 2026-07-17 12:56:20 +02:00
96 lines
3.0 KiB
Python
96 lines
3.0 KiB
Python
# coding: utf-8
|
|
|
|
"""
|
|
goose-server
|
|
|
|
An AI agent
|
|
|
|
The version of the OpenAPI document: 1.9.0
|
|
Contact: ai-oss-tools@block.xyz
|
|
Generated by OpenAPI Generator (https://openapi-generator.tech)
|
|
|
|
Do not edit the class manually.
|
|
""" # noqa: E501
|
|
|
|
|
|
import unittest
|
|
|
|
from goose_api.models.provider_metadata import ProviderMetadata
|
|
|
|
class TestProviderMetadata(unittest.TestCase):
|
|
"""ProviderMetadata unit test stubs"""
|
|
|
|
def setUp(self):
|
|
pass
|
|
|
|
def tearDown(self):
|
|
pass
|
|
|
|
def make_instance(self, include_optional) -> ProviderMetadata:
|
|
"""Test ProviderMetadata
|
|
include_optional is a boolean, when False only required
|
|
params are included, when True both required and
|
|
optional params are included """
|
|
# uncomment below to create an instance of `ProviderMetadata`
|
|
"""
|
|
model = ProviderMetadata()
|
|
if include_optional:
|
|
return ProviderMetadata(
|
|
config_keys = [
|
|
goose_api.models.config_key.ConfigKey(
|
|
default = '',
|
|
name = '',
|
|
oauth_flow = True,
|
|
required = True,
|
|
secret = True, )
|
|
],
|
|
default_model = '',
|
|
description = '',
|
|
display_name = '',
|
|
known_models = [
|
|
goose_api.models.model_info.ModelInfo(
|
|
context_limit = 0,
|
|
currency = '',
|
|
input_token_cost = 1.337,
|
|
name = '',
|
|
output_token_cost = 1.337,
|
|
supports_cache_control = True, )
|
|
],
|
|
model_doc_link = '',
|
|
name = ''
|
|
)
|
|
else:
|
|
return ProviderMetadata(
|
|
config_keys = [
|
|
goose_api.models.config_key.ConfigKey(
|
|
default = '',
|
|
name = '',
|
|
oauth_flow = True,
|
|
required = True,
|
|
secret = True, )
|
|
],
|
|
default_model = '',
|
|
description = '',
|
|
display_name = '',
|
|
known_models = [
|
|
goose_api.models.model_info.ModelInfo(
|
|
context_limit = 0,
|
|
currency = '',
|
|
input_token_cost = 1.337,
|
|
name = '',
|
|
output_token_cost = 1.337,
|
|
supports_cache_control = True, )
|
|
],
|
|
model_doc_link = '',
|
|
name = '',
|
|
)
|
|
"""
|
|
|
|
def testProviderMetadata(self):
|
|
"""Test ProviderMetadata"""
|
|
# inst_req_only = self.make_instance(include_optional=False)
|
|
# inst_req_and_optional = self.make_instance(include_optional=True)
|
|
|
|
if __name__ == '__main__':
|
|
unittest.main()
|