Merging PR_218 openai_rev package with new streamlit chat app
This commit is contained in:
@@ -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,35 @@
|
||||
# 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 typing import Optional
|
||||
|
||||
from selenium.webdriver.chromium.options import ChromiumOptions
|
||||
from selenium.webdriver.common.desired_capabilities import DesiredCapabilities
|
||||
|
||||
|
||||
class Options(ChromiumOptions):
|
||||
@property
|
||||
def default_capabilities(self) -> dict:
|
||||
return DesiredCapabilities.CHROME.copy()
|
||||
|
||||
def enable_mobile(
|
||||
self,
|
||||
android_package: str = "com.android.chrome",
|
||||
android_activity: Optional[str] = None,
|
||||
device_serial: Optional[str] = None,
|
||||
) -> None:
|
||||
super().enable_mobile(android_package, android_activity, device_serial)
|
||||
@@ -0,0 +1,52 @@
|
||||
# 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
|
||||
|
||||
from selenium.webdriver.chromium import service
|
||||
|
||||
DEFAULT_EXECUTABLE_PATH = "chromedriver"
|
||||
|
||||
|
||||
class Service(service.ChromiumService):
|
||||
"""A Service class that is responsible for the starting and stopping of
|
||||
`chromedriver`.
|
||||
|
||||
:param executable_path: install path of the chromedriver executable, defaults to `chromedriver`.
|
||||
:param port: Port for the service to run on, defaults to 0 where the operating system will decide.
|
||||
:param service_args: (Optional) List of args to be passed to the subprocess when launching the executable.
|
||||
:param log_path: (Optional) String to be passed to the executable as `--log-path`.
|
||||
: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,
|
||||
service_args: typing.Optional[typing.List[str]] = None,
|
||||
log_path: typing.Optional[str] = None,
|
||||
env: typing.Optional[typing.Mapping[str, str]] = None,
|
||||
**kwargs,
|
||||
) -> None:
|
||||
super().__init__(
|
||||
executable_path=executable_path,
|
||||
port=port,
|
||||
service_args=service_args,
|
||||
log_path=log_path,
|
||||
env=env,
|
||||
start_error_message="Please see https://chromedriver.chromium.org/home",
|
||||
**kwargs,
|
||||
)
|
||||
@@ -0,0 +1,97 @@
|
||||
# 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
|
||||
DEFAULT_KEEP_ALIVE = None
|
||||
|
||||
|
||||
class WebDriver(ChromiumDriver):
|
||||
"""Controls the ChromeDriver and allows you to drive the browser.
|
||||
|
||||
You will need to download the ChromeDriver executable from
|
||||
http://chromedriver.storage.googleapis.com/index.html
|
||||
"""
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
executable_path=DEFAULT_EXECUTABLE_PATH,
|
||||
port=DEFAULT_PORT,
|
||||
options: Options = None,
|
||||
service_args=None,
|
||||
desired_capabilities=None,
|
||||
service_log_path=DEFAULT_SERVICE_LOG_PATH,
|
||||
chrome_options=None,
|
||||
service: Service = None,
|
||||
keep_alive=DEFAULT_KEEP_ALIVE,
|
||||
) -> None:
|
||||
"""Creates a new instance of the chrome driver. Starts the service and
|
||||
then creates new instance of chrome 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 ChromeOptions
|
||||
- service - Service object for handling the browser driver if you need to pass extra details
|
||||
- service_args - Deprecated: List of args to pass to the driver service
|
||||
- desired_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.
|
||||
- keep_alive - Deprecated: Whether to configure ChromeRemoteConnection to use HTTP keep-alive.
|
||||
"""
|
||||
if executable_path != "chromedriver":
|
||||
warnings.warn(
|
||||
"executable_path has been deprecated, please pass in a Service object", DeprecationWarning, stacklevel=2
|
||||
)
|
||||
if chrome_options:
|
||||
warnings.warn("use options instead of chrome_options", DeprecationWarning, stacklevel=2)
|
||||
options = chrome_options
|
||||
if keep_alive != DEFAULT_KEEP_ALIVE:
|
||||
warnings.warn(
|
||||
"keep_alive has been deprecated, please pass in a Service object", DeprecationWarning, stacklevel=2
|
||||
)
|
||||
else:
|
||||
keep_alive = True
|
||||
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.CHROME["browserName"],
|
||||
"goog",
|
||||
port,
|
||||
options,
|
||||
service_args,
|
||||
desired_capabilities,
|
||||
service_log_path,
|
||||
service,
|
||||
keep_alive,
|
||||
)
|
||||
|
||||
def create_options(self) -> Options:
|
||||
return Options()
|
||||
Reference in New Issue
Block a user