Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
25 commits
Select commit Hold shift + click to select a range
b01d3e1
initial-commit-generated-code-no-manual-change
w-javed Jun 25, 2025
8b7f40f
cspell
w-javed Jun 25, 2025
8bbd8df
Adding scan_Session_id
Jun 26, 2025
6d85899
more changes
Jun 26, 2025
9796a9e
Add gitignore in the scan directory
Jun 26, 2025
1514e59
Update default timeout to 3600
Jun 26, 2025
e285869
updated client code
w-javed Jun 26, 2025
2afb610
Merge remote-tracking branch 'wjaved/gen_client_code_update' into ses…
Jun 26, 2025
effd181
cspell update
w-javed Jun 26, 2025
fa7901a
Merge remote-tracking branch 'wjaved/gen_client_code_update' into ses…
Jun 26, 2025
bde2bfc
updating client code
w-javed Jun 26, 2025
a803f93
Merge remote-tracking branch 'wjaved/gen_client_code_update' into ses…
Jun 26, 2025
2076065
Display name
Jun 26, 2025
aaaf152
Eval run has no atribute scan name
Jun 26, 2025
5edc399
Merge pull request #1 from nagkumar91/session_id_for_requests
w-javed Jun 26, 2025
b7774f6
updating client code
w-javed Jun 26, 2025
0bbc256
Remove the gitignore from being uploaded
Jun 26, 2025
3f4e0d1
Remove duplicate keys
Jun 26, 2025
32a993e
Make sure its still written
Jun 26, 2025
8899136
Merge pull request #2 from nagkumar91/session_id_for_requests
w-javed Jun 26, 2025
1f0650c
Change the client to be used to extract responses for multiturn
Jun 27, 2025
f0f9b74
remove the ununsed init
Jun 27, 2025
c7205e3
Merge pull request #3 from nagkumar91/session_id_for_requests
w-javed Jun 27, 2025
17ddf0f
quick typo fix
w-javed Jun 30, 2025
3ada0c9
black-update
w-javed Jun 30, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Original file line number Diff line number Diff line change
@@ -1,32 +1,32 @@
# coding=utf-8
# --------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for license information.
# Code generated by Microsoft (R) Python Code Generator.
# Changes may cause incorrect behavior and will be lost if the code is regenerated.
# --------------------------------------------------------------------------
# pylint: disable=wrong-import-position
from typing import TYPE_CHECKING
if TYPE_CHECKING:
from ._patch import * # pylint: disable=unused-wildcard-import
from ._client import AIProjectClient # type: ignore
from ._version import VERSION
__version__ = VERSION
try:
from ._patch import __all__ as _patch_all
from ._patch import *
except ImportError:
_patch_all = []
from ._patch import patch_sdk as _patch_sdk
__all__ = [
"AIProjectClient",
]
__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore
_patch_sdk()
# coding=utf-8
# --------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for license information.
# Code generated by Microsoft (R) Python Code Generator.
# Changes may cause incorrect behavior and will be lost if the code is regenerated.
# --------------------------------------------------------------------------
# pylint: disable=wrong-import-position

from typing import TYPE_CHECKING

if TYPE_CHECKING:
from ._patch import * # pylint: disable=unused-wildcard-import

from ._client import AIProjectClient # type: ignore
from ._version import VERSION

__version__ = VERSION

try:
from ._patch import __all__ as _patch_all
from ._patch import *
except ImportError:
_patch_all = []
from ._patch import patch_sdk as _patch_sdk

__all__ = [
"AIProjectClient",
]
__all__.extend([p for p in _patch_all if p not in __all__]) # pyright: ignore

_patch_sdk()
Original file line number Diff line number Diff line change
@@ -1,139 +1,136 @@
# coding=utf-8
# --------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for license information.
# Code generated by Microsoft (R) Python Code Generator.
# Changes may cause incorrect behavior and will be lost if the code is regenerated.
# --------------------------------------------------------------------------

from copy import deepcopy
from typing import Any, TYPE_CHECKING, Union
from typing_extensions import Self

from azure.core import PipelineClient
from azure.core.credentials import AzureKeyCredential
from azure.core.pipeline import policies
from azure.core.rest import HttpRequest, HttpResponse

from ._configuration import AIProjectClientConfiguration
from ._serialization import Deserializer, Serializer
from .operations import (
ConnectionsOperations,
DatasetsOperations,
DeploymentsOperations,
EvaluationResultsOperations,
EvaluationsOperations,
IndexesOperations,
RedTeamsOperations,
ServicePatternsOperations,
)

if TYPE_CHECKING:
from azure.core.credentials import TokenCredential


class AIProjectClient: # pylint: disable=too-many-instance-attributes
"""AIProjectClient.

:ivar service_patterns: ServicePatternsOperations operations
:vartype service_patterns: azure.ai.projects.onedp.operations.ServicePatternsOperations
:ivar connections: ConnectionsOperations operations
:vartype connections: azure.ai.projects.onedp.operations.ConnectionsOperations
:ivar evaluations: EvaluationsOperations operations
:vartype evaluations: azure.ai.projects.onedp.operations.EvaluationsOperations
:ivar datasets: DatasetsOperations operations
:vartype datasets: azure.ai.projects.onedp.operations.DatasetsOperations
:ivar indexes: IndexesOperations operations
:vartype indexes: azure.ai.projects.onedp.operations.IndexesOperations
:ivar deployments: DeploymentsOperations operations
:vartype deployments: azure.ai.projects.onedp.operations.DeploymentsOperations
:ivar red_teams: RedTeamsOperations operations
:vartype red_teams: azure.ai.projects.onedp.operations.RedTeamsOperations
:ivar evaluation_results: EvaluationResultsOperations operations
:vartype evaluation_results: azure.ai.projects.onedp.operations.EvaluationResultsOperations
:param endpoint: Project endpoint in the form of:
https://<aiservices-id>.services.ai.azure.com/api/projects/<project-name>. Required.
:type endpoint: str
:param credential: Credential used to authenticate requests to the service. Is either a key
credential type or a token credential type. Required.
:type credential: ~azure.core.credentials.AzureKeyCredential or
~azure.core.credentials.TokenCredential
:keyword api_version: The API version to use for this operation. Default value is
"2025-05-15-preview". Note that overriding this default value may result in unsupported
behavior.
:paramtype api_version: str
"""

def __init__(self, endpoint: str, credential: Union[AzureKeyCredential, "TokenCredential"], **kwargs: Any) -> None:
_endpoint = "{endpoint}"
self._config = AIProjectClientConfiguration(endpoint=endpoint, credential=credential, **kwargs)
_policies = kwargs.pop("policies", None)
if _policies is None:
_policies = [
policies.RequestIdPolicy(**kwargs),
self._config.headers_policy,
self._config.user_agent_policy,
self._config.proxy_policy,
policies.ContentDecodePolicy(**kwargs),
self._config.redirect_policy,
self._config.retry_policy,
self._config.authentication_policy,
self._config.custom_hook_policy,
self._config.logging_policy,
policies.DistributedTracingPolicy(**kwargs),
policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,
self._config.http_logging_policy,
]
self._client: PipelineClient = PipelineClient(base_url=_endpoint, policies=_policies, **kwargs)

self._serialize = Serializer()
self._deserialize = Deserializer()
self._serialize.client_side_validation = False
self.service_patterns = ServicePatternsOperations(
self._client, self._config, self._serialize, self._deserialize
)
self.connections = ConnectionsOperations(self._client, self._config, self._serialize, self._deserialize)
self.evaluations = EvaluationsOperations(self._client, self._config, self._serialize, self._deserialize)
self.datasets = DatasetsOperations(self._client, self._config, self._serialize, self._deserialize)
self.indexes = IndexesOperations(self._client, self._config, self._serialize, self._deserialize)
self.deployments = DeploymentsOperations(self._client, self._config, self._serialize, self._deserialize)
self.red_teams = RedTeamsOperations(self._client, self._config, self._serialize, self._deserialize)
self.evaluation_results = EvaluationResultsOperations(
self._client, self._config, self._serialize, self._deserialize
)

def send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse:
"""Runs the network request through the client's chained policies.

>>> from azure.core.rest import HttpRequest
>>> request = HttpRequest("GET", "https://www.example.org/")
<HttpRequest [GET], url: 'https://www.example.org/'>
>>> response = client.send_request(request)
<HttpResponse: 200 OK>

For more information on this code flow, see https://aka.ms/azsdk/dpcodegen/python/send_request

:param request: The network request you want to make. Required.
:type request: ~azure.core.rest.HttpRequest
:keyword bool stream: Whether the response payload will be streamed. Defaults to False.
:return: The response of your network call. Does not do error handling on your response.
:rtype: ~azure.core.rest.HttpResponse
"""

request_copy = deepcopy(request)
path_format_arguments = {
"endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True),
}

request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments)
return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore

def close(self) -> None:
self._client.close()

def __enter__(self) -> Self:
self._client.__enter__()
return self

def __exit__(self, *exc_details: Any) -> None:
self._client.__exit__(*exc_details)
# coding=utf-8
# --------------------------------------------------------------------------
# Copyright (c) Microsoft Corporation. All rights reserved.
# Licensed under the MIT License. See License.txt in the project root for license information.
# Code generated by Microsoft (R) Python Code Generator.
# Changes may cause incorrect behavior and will be lost if the code is regenerated.
# --------------------------------------------------------------------------

from copy import deepcopy
from typing import Any, TYPE_CHECKING
from typing_extensions import Self

from azure.core import PipelineClient
from azure.core.pipeline import policies
from azure.core.rest import HttpRequest, HttpResponse

from ._configuration import AIProjectClientConfiguration
from ._utils.serialization import Deserializer, Serializer
from .operations import (
ConnectionsOperations,
DatasetsOperations,
DeploymentsOperations,
EvaluationResultsOperations,
EvaluationsOperations,
IndexesOperations,
RedTeamsOperations,
)

if TYPE_CHECKING:
from azure.core.credentials import TokenCredential


class AIProjectClient: # pylint: disable=too-many-instance-attributes
"""AIProjectClient.

:ivar connections: ConnectionsOperations operations
:vartype connections: azure.ai.projects.onedp.operations.ConnectionsOperations
:ivar evaluations: EvaluationsOperations operations
:vartype evaluations: azure.ai.projects.onedp.operations.EvaluationsOperations
:ivar datasets: DatasetsOperations operations
:vartype datasets: azure.ai.projects.onedp.operations.DatasetsOperations
:ivar indexes: IndexesOperations operations
:vartype indexes: azure.ai.projects.onedp.operations.IndexesOperations
:ivar deployments: DeploymentsOperations operations
:vartype deployments: azure.ai.projects.onedp.operations.DeploymentsOperations
:ivar red_teams: RedTeamsOperations operations
:vartype red_teams: azure.ai.projects.onedp.operations.RedTeamsOperations
:ivar evaluation_results: EvaluationResultsOperations operations
:vartype evaluation_results: azure.ai.projects.onedp.operations.EvaluationResultsOperations
:param endpoint: Project endpoint. In the form
"https://<your-ai-services-account-name>.services.ai.azure.com/api/projects/_project"
if your Foundry Hub has only one Project, or to use the default Project in your Hub. Or in the
form
"https://<your-ai-services-account-name>.services.ai.azure.com/api/projects/<your-project-name>"
if you want to explicitly
specify the Foundry Project name. Required.
:type endpoint: str
:param credential: Credential used to authenticate requests to the service. Required.
:type credential: ~azure.core.credentials.TokenCredential
:keyword api_version: The API version to use for this operation. Default value is
"2025-05-15-preview". Note that overriding this default value may result in unsupported
behavior.
:paramtype api_version: str
"""

def __init__(self, endpoint: str, credential: "TokenCredential", **kwargs: Any) -> None:
_endpoint = "{endpoint}"
self._config = AIProjectClientConfiguration(endpoint=endpoint, credential=credential, **kwargs)

_policies = kwargs.pop("policies", None)
if _policies is None:
_policies = [
policies.RequestIdPolicy(**kwargs),
self._config.headers_policy,
self._config.user_agent_policy,
self._config.proxy_policy,
policies.ContentDecodePolicy(**kwargs),
self._config.redirect_policy,
self._config.retry_policy,
self._config.authentication_policy,
self._config.custom_hook_policy,
self._config.logging_policy,
policies.DistributedTracingPolicy(**kwargs),
policies.SensitiveHeaderCleanupPolicy(**kwargs) if self._config.redirect_policy else None,
self._config.http_logging_policy,
]
self._client: PipelineClient = PipelineClient(base_url=_endpoint, policies=_policies, **kwargs)

self._serialize = Serializer()
self._deserialize = Deserializer()
self._serialize.client_side_validation = False
self.connections = ConnectionsOperations(self._client, self._config, self._serialize, self._deserialize)
self.evaluations = EvaluationsOperations(self._client, self._config, self._serialize, self._deserialize)
self.datasets = DatasetsOperations(self._client, self._config, self._serialize, self._deserialize)
self.indexes = IndexesOperations(self._client, self._config, self._serialize, self._deserialize)
self.deployments = DeploymentsOperations(self._client, self._config, self._serialize, self._deserialize)
self.red_teams = RedTeamsOperations(self._client, self._config, self._serialize, self._deserialize)
self.evaluation_results = EvaluationResultsOperations(
self._client, self._config, self._serialize, self._deserialize
)

def send_request(self, request: HttpRequest, *, stream: bool = False, **kwargs: Any) -> HttpResponse:
"""Runs the network request through the client's chained policies.

>>> from azure.core.rest import HttpRequest
>>> request = HttpRequest("GET", "https://www.example.org/")
<HttpRequest [GET], url: 'https://www.example.org/'>
>>> response = client.send_request(request)
<HttpResponse: 200 OK>

For more information on this code flow, see https://aka.ms/azsdk/dpcodegen/python/send_request

:param request: The network request you want to make. Required.
:type request: ~azure.core.rest.HttpRequest
:keyword bool stream: Whether the response payload will be streamed. Defaults to False.
:return: The response of your network call. Does not do error handling on your response.
:rtype: ~azure.core.rest.HttpResponse
"""

request_copy = deepcopy(request)
path_format_arguments = {
"endpoint": self._serialize.url("self._config.endpoint", self._config.endpoint, "str", skip_quote=True),
}

request_copy.url = self._client.format_url(request_copy.url, **path_format_arguments)
return self._client.send_request(request_copy, stream=stream, **kwargs) # type: ignore

def close(self) -> None:
self._client.close()

def __enter__(self) -> Self:
self._client.__enter__()
return self

def __exit__(self, *exc_details: Any) -> None:
self._client.__exit__(*exc_details)
Loading