Skip to content

Commit

Permalink
modify test
Browse files Browse the repository at this point in the history
  • Loading branch information
@Walter-jin
Walter-jin committed Aug 29, 2024
1 parent b5a8dd5 commit 9f9c4f2
Show file tree
Hide file tree
Showing 3 changed files with 13 additions and 27 deletions.
25 changes: 7 additions & 18 deletions api/tests/integration_tests/model_runtime/oci/test_llm.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -22,19 +22,16 @@ def test_validate_credentials():
model.validate_credentials(
model='cohere.command-r-plus',
credentials={
'oci_api_profile': 'invalid_key',
'secret_id': 'invalid_key',
'secret_key': 'invalid_key'
'oci_config_content': 'invalid_key',
'oci_key_content': 'invalid_key'
}
)

model.validate_credentials(
model='cohere.command-r-plus',
credentials={
'oci_config_content': os.environ.get('OCI_CONFIG_CONTENT'),
'oci_key_content': os.environ.get('OCI_KEY_CONTENT'),
'secret_id': os.environ.get('SECRET_ID'),
'secret_key': os.environ.get('SECRET_KEY')
'oci_key_content': os.environ.get('OCI_KEY_CONTENT')
}
)

Expand All @@ -46,9 +43,7 @@ def test_invoke_model():
model='cohere.command-r-plus',
credentials={
'oci_config_content': os.environ.get('OCI_CONFIG_CONTENT'),
'oci_key_content': os.environ.get('OCI_KEY_CONTENT'),
'secret_id': os.environ.get('SECRET_ID'),
'secret_key': os.environ.get('SECRET_KEY')
'oci_key_content': os.environ.get('OCI_KEY_CONTENT')
},
prompt_messages=[
UserPromptMessage(
Expand All @@ -75,9 +70,7 @@ def test_invoke_stream_model():
model='meta.llama-3-70b-instruct',
credentials={
'oci_config_content': os.environ.get('OCI_CONFIG_CONTENT'),
'oci_key_content': os.environ.get('OCI_KEY_CONTENT'),
'secret_id': os.environ.get('SECRET_ID'),
'secret_key': os.environ.get('SECRET_KEY')
'oci_key_content': os.environ.get('OCI_KEY_CONTENT')
},
prompt_messages=[
UserPromptMessage(
Expand Down Expand Up @@ -109,9 +102,7 @@ def test_invoke_model_with_function():
model='cohere.command-r-plus',
credentials={
'oci_config_content': os.environ.get('OCI_CONFIG_CONTENT'),
'oci_key_content': os.environ.get('OCI_KEY_CONTENT'),
'secret_id': os.environ.get('SECRET_ID'),
'secret_key': os.environ.get('SECRET_KEY')
'oci_key_content': os.environ.get('OCI_KEY_CONTENT')
},
prompt_messages=[
UserPromptMessage(
Expand Down Expand Up @@ -160,9 +151,7 @@ def test_get_num_tokens():
model='cohere.command-r-plus',
credentials={
'oci_config_content': os.environ.get('OCI_CONFIG_CONTENT'),
'oci_key_content': os.environ.get('OCI_KEY_CONTENT'),
'secret_id': os.environ.get('SECRET_ID'),
'secret_key': os.environ.get('SECRET_KEY')
'oci_key_content': os.environ.get('OCI_KEY_CONTENT')
},
prompt_messages=[
SystemPromptMessage(
Expand Down
2 changes: 1 addition & 1 deletion api/tests/integration_tests/model_runtime/oci/test_provider.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,6 @@ def test_validate_provider_credentials():
provider.validate_provider_credentials(
credentials={
'oci_config_content': os.environ.get('OCI_CONFIG_CONTENT'),
'oci_key_content': os.environ.get('OCI_KEY_CONTENT'),
'oci_key_content': os.environ.get('OCI_KEY_CONTENT')
}
)
13 changes: 5 additions & 8 deletions api/tests/integration_tests/model_runtime/oci/test_text_embedding.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -14,17 +14,16 @@ def test_validate_credentials():
model.validate_credentials(
model='cohere.embed-multilingual-v3.0',
credentials={
'api_key': 'invalid_key',
'oci_api_profile': 'invalid_key'
'oci_config_content': 'invalid_key',
'oci_key_content': 'invalid_key'
}
)

model.validate_credentials(
model='cohere.embed-multilingual-v3.0',
credentials={
'oci_config_content': os.environ.get('OCI_CONFIG_CONTENT'),
'oci_key_content': os.environ.get('OCI_KEY_CONTENT'),
'oci_api_profile': os.environ.get('OCI_API_PROFILE', 'GENERATEAI')
'oci_key_content': os.environ.get('OCI_KEY_CONTENT')
}
)

Expand All @@ -36,8 +35,7 @@ def test_invoke_model():
model='cohere.embed-multilingual-v3.0',
credentials={
'oci_config_content': os.environ.get('OCI_CONFIG_CONTENT'),
'oci_key_content': os.environ.get('OCI_KEY_CONTENT'),
'oci_api_profile': os.environ.get('OCI_API_PROFILE')
'oci_key_content': os.environ.get('OCI_KEY_CONTENT')
},
texts=[
"hello",
Expand All @@ -60,8 +58,7 @@ def test_get_num_tokens():
model='cohere.embed-multilingual-v3.0',
credentials={
'oci_config_content': os.environ.get('OCI_CONFIG_CONTENT'),
'oci_key_content': os.environ.get('OCI_KEY_CONTENT'),
'oci_api_profile': os.environ.get('OCI_API_PROFILE')
'oci_key_content': os.environ.get('OCI_KEY_CONTENT')
},
texts=[
"hello",
Expand Down

0 comments on commit 9f9c4f2

Please sign in to comment.