Merging PR_218 openai_rev package with new streamlit chat app
This commit is contained in:
noptuno
@@ -0,0 +1,16 @@
|
||||
# Licensed to the Software Freedom Conservancy (SFC) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The SFC licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing,
|
||||
# software distributed under the License is distributed on an
|
||||
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
||||
# KIND, either express or implied. See the License for the
|
||||
# specific language governing permissions and limitations
|
||||
# under the License.
|
||||
@@ -0,0 +1,50 @@
|
||||
# Licensed to the Software Freedom Conservancy (SFC) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The SFC licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing,
|
||||
# software distributed under the License is distributed on an
|
||||
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
||||
# KIND, either express or implied. See the License for the
|
||||
# specific language governing permissions and limitations
|
||||
# under the License.
|
||||
|
||||
from selenium.webdriver.chromium.options import ChromiumOptions
|
||||
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
|
||||
|
||||
|
||||
class Options(ChromiumOptions):
|
||||
KEY = "ms:edgeOptions"
|
||||
|
||||
def __init__(self) -> None:
|
||||
super().__init__()
|
||||
self._use_webview = False
|
||||
|
||||
@property
|
||||
def use_webview(self) -> bool:
|
||||
return self._use_webview
|
||||
|
||||
@use_webview.setter
|
||||
def use_webview(self, value: bool) -> None:
|
||||
self._use_webview = bool(value)
|
||||
|
||||
def to_capabilities(self) -> dict:
|
||||
"""
|
||||
Creates a capabilities with all the options that have been set and
|
||||
:Returns: A dictionary with everything
|
||||
"""
|
||||
caps = super().to_capabilities()
|
||||
if self._use_webview:
|
||||
caps["browserName"] = "webview2"
|
||||
|
||||
return caps
|
||||
|
||||
@property
|
||||
def default_capabilities(self) -> dict:
|
||||
return DesiredCapabilities.EDGE.copy()
|
||||
@@ -0,0 +1,66 @@
|
||||
# Licensed to the Software Freedom Conservancy (SFC) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The SFC licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing,
|
||||
# software distributed under the License is distributed on an
|
||||
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
||||
# KIND, either express or implied. See the License for the
|
||||
# specific language governing permissions and limitations
|
||||
# under the License.
|
||||
import typing
|
||||
import warnings
|
||||
|
||||
from selenium.webdriver.chromium import service
|
||||
|
||||
DEFAULT_EXECUTABLE_PATH = "msedgedriver"
|
||||
|
||||
|
||||
class Service(service.ChromiumService):
|
||||
"""A Service class that is responsible for the starting and stopping of
|
||||
`msedgedriver`.
|
||||
|
||||
:param executable_path: install path of the msedgedriver executable, defaults to `msedgedriver`.
|
||||
:param port: Port for the service to run on, defaults to 0 where the operating system will decide.
|
||||
:param verbose: (Deprecated) Whether to make the webdriver more verbose (passes the --verbose option to the binary).
|
||||
Defaults to False.
|
||||
:param log_path: (Optional) String to be passed to the executable as `--log-path`.
|
||||
:param service_args: (Optional) List of args to be passed to the subprocess when launching the executable.
|
||||
:param env: (Optional) Mapping of environment variables for the new process, defaults to `os.environ`.
|
||||
"""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
executable_path: str = DEFAULT_EXECUTABLE_PATH,
|
||||
port: int = 0,
|
||||
verbose: bool = False,
|
||||
log_path: typing.Optional[str] = None,
|
||||
service_args: typing.Optional[typing.List[str]] = None,
|
||||
env: typing.Optional[typing.Mapping[str, str]] = None,
|
||||
**kwargs,
|
||||
) -> None:
|
||||
self.service_args = service_args or []
|
||||
|
||||
if verbose:
|
||||
warnings.warn(
|
||||
"verbose=True is deprecated. Use `service_args=['--verbose', ...]` instead.",
|
||||
DeprecationWarning,
|
||||
stacklevel=2,
|
||||
)
|
||||
self.service_args.append("--verbose")
|
||||
|
||||
super().__init__(
|
||||
executable_path=executable_path,
|
||||
port=port,
|
||||
service_args=service_args,
|
||||
log_path=log_path,
|
||||
env=env,
|
||||
start_error_message="Please download from https://developer.microsoft.com/en-us/microsoft-edge/tools/webdriver/",
|
||||
**kwargs,
|
||||
)
|
||||
@@ -0,0 +1,89 @@
|
||||
# Licensed to the Software Freedom Conservancy (SFC) under one
|
||||
# or more contributor license agreements. See the NOTICE file
|
||||
# distributed with this work for additional information
|
||||
# regarding copyright ownership. The SFC licenses this file
|
||||
# to you under the Apache License, Version 2.0 (the
|
||||
# "License"); you may not use this file except in compliance
|
||||
# with the License. You may obtain a copy of the License at
|
||||
#
|
||||
# http://www.apache.org/licenses/LICENSE-2.0
|
||||
#
|
||||
# Unless required by applicable law or agreed to in writing,
|
||||
# software distributed under the License is distributed on an
|
||||
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
||||
# KIND, either express or implied. See the License for the
|
||||
# specific language governing permissions and limitations
|
||||
# under the License.
|
||||
import warnings
|
||||
|
||||
from selenium.webdriver.chromium.webdriver import ChromiumDriver
|
||||
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
|
||||
from selenium.webdriver.common.driver_finder import DriverFinder
|
||||
|
||||
from .options import Options
|
||||
from .service import DEFAULT_EXECUTABLE_PATH
|
||||
from .service import Service
|
||||
|
||||
DEFAULT_PORT = 0
|
||||
DEFAULT_SERVICE_LOG_PATH = None
|
||||
|
||||
|
||||
class WebDriver(ChromiumDriver):
|
||||
"""Controls the Microsoft Edge driver and allows you to drive the browser.
|
||||
|
||||
You will need to download the MSEdgeDriver executable from
|
||||
https://developer.microsoft.com/microsoft-edge/tools/webdriver/
|
||||
"""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
executable_path=DEFAULT_EXECUTABLE_PATH,
|
||||
port=DEFAULT_PORT,
|
||||
options: Options = Options(),
|
||||
service_args=None,
|
||||
capabilities=None,
|
||||
service_log_path=DEFAULT_SERVICE_LOG_PATH,
|
||||
service: Service = None,
|
||||
keep_alive=False,
|
||||
verbose=False, # Todo: Why is this now unused?
|
||||
) -> None:
|
||||
"""Creates a new instance of the edge driver. Starts the service and
|
||||
then creates new instance of edge driver.
|
||||
|
||||
:Args:
|
||||
- executable_path - Deprecated: path to the executable. If the default is used it assumes the executable is in the $PATH
|
||||
- port - Deprecated: port you would like the service to run, if left as 0, a free port will be found.
|
||||
- options - this takes an instance of EdgeOptions
|
||||
- service_args - Deprecated: List of args to pass to the driver service
|
||||
- capabilities - Deprecated: Dictionary object with non-browser specific
|
||||
capabilities only, such as "proxy" or "loggingPref".
|
||||
- service_log_path - Deprecated: Where to log information from the driver.
|
||||
- service - Service object for handling the browser driver if you need to pass extra details
|
||||
- keep_alive - Whether to configure EdgeRemoteConnection to use HTTP keep-alive.
|
||||
- verbose - whether to set verbose logging in the service.
|
||||
"""
|
||||
if executable_path != "msedgedriver":
|
||||
warnings.warn(
|
||||
"executable_path has been deprecated, please pass in a Service object", DeprecationWarning, stacklevel=2
|
||||
)
|
||||
|
||||
if not options:
|
||||
options = self.create_options()
|
||||
if not service:
|
||||
service = Service(executable_path, port, service_args, service_log_path)
|
||||
service.path = DriverFinder.get_path(service, options)
|
||||
|
||||
super().__init__(
|
||||
DesiredCapabilities.EDGE["browserName"],
|
||||
"ms",
|
||||
port,
|
||||
options,
|
||||
service_args,
|
||||
capabilities,
|
||||
service_log_path,
|
||||
service,
|
||||
keep_alive,
|
||||
)
|
||||
|
||||
def create_options(self) -> Options:
|
||||
return Options()
|
||||
Reference in New Issue
Block a user