Skip to main content

Cogment python SDK

Project description


Latest final release Apache 2 License Changelog

Cogment is an innovative open source AI platform designed to leverage the advent of AI to benefit humankind through human-AI collaboration developed by AI Redefined. Cogment enables AI researchers and engineers to build, train and operate AI agents in simulated or real environments shared with humans. For the full user documentation visit

This module, cogment-py-sdk, is the Python SDK for making use of Cogment when working with Python. It's full documentation can be consulted at



Make sure you have the following installed:

Install dependencies

We recommend using a virtualenv to isolate your dependencies. On most unix flavors you create and activate this by running the following:

$ python -m venv .venv
$ source .venv/bin/activate

once you are finished with the virtualenv you can deactivate it by running:

$ deactivate

Install the dependencies, including downloading and building the cogment protobuf API by running the following:

$ pip install -r requirements.txt
$ python -m setup develop

The second line will call the following commands to deal with the cogment API (those can also be called seprately)

$ python -m setup retrieve_cogment_api
$ python -m setup build_cogment_api_protos

Define used Cogment protobuf API

The version of the used cogment protobuf API is defined in the .cogment-api.yaml file at the root of the repository. The following can be defined:

  • cogment_version: "latest", is the default, it retrieves the api from the latest Cogment release (excluding pre-releases),
  • cogment_version: "vMAJOR.MINOR.PATCH[-PRERELEASE]", retrieves the api from any Cogment release.
  • cogment_api_path: "../RELATIVE/PATH/TO/LOCAL/COGMENT/INSTALL/include/cogment/api", retrieves a local version of the api found at the given path (e.g. common.proto should be at ${cogment_api_path}/common.proto); if set, this overrides cogment_version.

After any change, do not forget to run pip install -e . for it to be taken into account.


Integration tests

These tests launch and use Cogment, by default they'll use they'll download and use the latest released version of Cogment.

$ pytest --launch-orchestrator

The following environment can be defined to change this behavior, either directly in the terminal or in a .env file located at the root of the repository:

COGMENT_PATH="/path/to/cogment" # local path to cogment binary
COGMENT_VERSION="v2.2.0" # cogment version to download


Run the pycodestyle using

$ pycodestyle

Run the mypy type annotations checks using

$ mypy .

Check conflicting dependencies with "popular" Python packages


This script will check for conflicts required by the cogment-py-sdk and the popular Python packages in the AI/ML/Data ecosystem.

Build the source package

Build the source package

python -m build

Release process

People having mainteners rights of the repository can follow these steps to release a version MAJOR.MINOR.PATCH. The versioning scheme follows Semantic Versioning.

  1. Run ./scripts/ MAJOR.MINOR.PATCH to create the release branch and update the version of the package,
  2. On the release branch, check and update the changelog if needed,
  3. Make sure ./.cogment-api.yaml specifies fixed version to ensure rebuildability,
  4. Make sure everything's fine on CI,
  5. Run ./scripts/ MAJOR.MINOR.PATCH to create the specific version section in the changelog, merge the release branch in main, create the release tag and update the develop branch with those.

The rest, publishing the package to PyPI and updating the mirror repositories, is handled directly by the CI.


All notable changes to this project will be documented in this file.

The format is based on Keep a Changelog and this project adheres to Semantic Versioning.


v2.10.1 - 2024-01-06


  • Updates grpcio to work with python 3.10 on macOS.

v2.10.0 - 2023-11-28


  • Ability to define metadata in the Context constructor, they will be passed to all grpc calls.
  • Support for SSL grpc endpoints without a private key.


  • context.get_controller([...]), context.get_datastore([...]), context.get_model_registry_v2([...]) can always be awaited.

v2.9.0 - 2023-11-14


  • 'Status' RPC support for all services


  • Missing directory service type in API

v2.8.3 - 2023-08-22


  • Upgraded to support PyYaml up to 6.0.1 as v5 was broken by the release of cython 3.

v2.8.2 - 2023-07-10


  • Return None when no data is availble in datastore content

v2.8.1 - 2023-06-06


  • Use the most recent directory entry when there are duplicates
  • Self IP address discovery functionality is now working on macOS

v2.8.0 - 2023-04-21


  • Option to request full info (not just id and state) from 'watch_trials'
  • Ability to use the SDK without spec (cog_settings)
  • Automatic port selection for 'Context.serve_all_registered()'
  • Ability to specify an explicit host to register a service to the directory
  • Ability to specify trial properties for 'Datastore.all_trials()' and 'Datastore.get_trials()'
  • Ability to specify trial ids for 'Datastore.all_trials()'


  • Enable warning and error logging even if user did not define a logging handler
  • Warn user when no logging handler is defined (i.e. proper logging is disabled)
  • Better self IP address discovery functionality

v2.7.1 - 2023-03-06


  • Deadlock in 'wait_for_newer' method of LatestModel

v2.7.0 - 2023-02-28


  • Facility to track latest models automatically in the Model Registry


  • 'properties' attribute to TrialParameters and TrialInfo

v2.6.0 - 2023-02-03


  • Prometheus service is disabled by default


  • Model Registry method iteration_updates to stream/push new model iterations
  • Parameter wait_for_trials (for Datastore method all_trials) to wait for new trials

v2.5.0 - 2023-01-26


  • In Datastore reward could be returned as 'None' instead of 0.0


  • Simplified interface for Model Registry (context.get_model_registry_v2)

v2.4.1 - 2022-12-02

  • Support and dependency check for python 3.10

v2.4.0 - 2022-11-18


  • 'COGMENT_LOG_LEVEL' environment variable
  • Model Registry client


  • Port parameter needs to be an int
  • Context datastore request with default endpoint

v2.3.0 - 2022-09-20


  • Reward tick ID was hard coded to -1


  • Deprecation warning are only logged once


  • 'Version' grpc method for pre-trial hooks
  • Out-of-sync flag in datalog samples
  • Parameter 'nb_buffered_ticks'
  • Support for actor disconnection and response timeout parameters
  • Global Directory integration
  • Support for self registration to the Directory
  • Use environment variable for Directory access details

v2.2.2 - 2022-06-20


  • Generate to handle multiple imports properly
  • Output deprecated warning for use of 'snapshot' in received observation

v2.2.1 - 2022-05-27


  • Fix install of the package from a wheel


  • Replace poetry centric build process with modern setuptools

v2.2.0 - 2022-05-07


  • Introduce cogment.generate.generate, a function making the code generation previously available through python -m cogment.generate accessible as a function.


  • Integration tests are now able to install the Cogment CLI on their own.
  • Integration tests no longer uses a Docker image.


  • Fix integration test junit report output.

v2.1.1 - 2022-04-11


  • Fixed a typo in the TrialParameter "environemnt_name" to "environment_name"

v2.1.0 - 2022-03-25


  • Update of gRPC version to >=1.42 & <1.45
  • Deprecate cogment.LogParams
  • Deprecate current way to use PrehookSession
  • Deprecate Session 'event_loop', renamed to 'all_events'
  • Deprecate DatalogSession 'get_all_samples' renamed 'all_samples'
  • Standardize string ouput of classes


  • cogment.TrialParameters and cogment.ActorParameters
  • Add ability to provide parameters on trial start call
  • Datastore SDK


  • Fix the failure of the cogment package caused by the partial removal of gRPC 1.45 from pypi

v2.0.2 - 2022-01-03

v2.0.1 - 2022-01-03

v2.0.0 - 2022-01-03

v2.0.0-rc3 - 2021-12-16


  • cogment.generate now requires semantic arguments

v2.0.0-rc2 - 2021-12-10


  • Cleanup user output
  • Require grpc protocol url for endpoints

v2.0.0-rc1 - 2021-12-01


  • Major internal changes to match API 2.0
  • Minor use changes to match API 2.0 and improve usability

v1.3.1 - 2021-11-18


  • Restrict the SDK to python >= 3.7 and <3.10 to maintain compatiblity with tensorflow ^2.7.0.
  • Update the cogment api to v1.0.0 to v1.2.1


  • Added expanation for strict 1.38 requirement on grpcio-tools.
  • Enable pre-trial hooks to set a trial's environment's implementation in prehook_session.environment_implementation.

v1.3.0 - 2021-09-23


  • Add python -m cogment.generate to compile the protobuf files and generate for a cogment project. This is designed as a replacement for the cogment generate command for python cogment components.


  • Update grpc version used to 1.38.1
  • Better management of exception in user implementation coroutine

v1.2.0 - 2021-06-23


  • The default Prometheus registry is now used by default, it was previously a custom built registry. Furthermore, any prometheus registry can be passed to the constructor for cogment.Context. If None is passed prometheus metrics are disabled entirely.
  • Passing None as the prometheus_port in the constructor for cogment.Context disables the launch of the Prometheus server.

v1.1.1 - 2021-06-17


  • Update copyright notice to use the legal name of AI Redefined Inc.


  • Exit datalog loop on receiving the trial ended state
  • Simulatenous trials with client actors
  • Fix index in RecvAction to match the actor list

v1.1.0 - 2021-06-04


  • Internal update to change assert into more useful statements
  • Internal update to add __str__ to user accessible classes
  • Improve log output
  • Improve/add error reporting


  • Add function get_remote_versions to controller retrieving the remote versions including api and orchestrator versions.

v1.0.0 - 2021-05-10

  • Initial public release.



  • Environment can now receive messages

v1.0.0-beta3 - 2021-04-26


  • Breaking Change Fix Controller.get_actors(), now properly retrieves the actor in a given trial, the function is now async.
  • Fix unecessary exception thrown in watch_trial when async is cancelled


  • Add Controller.get_trial_info() to retrieve the information of a given trial.

v1.0.0-beta2 - 2021-04-15


  • Add a bit of code to make sure we are running in an asyncio task
  • EventType.FINAL events do not contain data anymore
  • Fix when replies are None
  • Add raw_trial_params in the datalog session


  • actor implementations metrics are now reported under impl_name

v1.0.0-beta1 - 2021-04-07

  • Initial beta release, no more breaking changes should be introduced.

v1.0.0-alpha12 - 2021-04-01


  • Rename ActorClass id to name


  • Remove unused feedback_space parameters from ActorClass.

v1.0.0-alpha11 - 2021-03-30

  • Technical release, updating dependencies to fixed versions.

v1.0.0-alpha10 - 2021-03-30


  • Log exporter is now available: trials param, observations, rewards, messages of every trials are sent to the log exporter service.


  • The value returned from a RecvReward.all_source is now a class RecvRewardSource instead of a tuple
  • Removed parameter user_id from start_trial (it was not necessary)

v1.0.0-alpha9 - 2021-03-10


  • Breaking Introduce Controller, built for a given orchestrator endpoint it can be used to start and terminate trials. Previous ways of accessing the same such are discontinued.
  • Introduce the ability to listen for trial events (startup, ending, ...) using Controller.watch_trials.

v1.0.0-alpha8 - 2021-02-26


  • Session event loops now return classes

v1.0.0-alpha7 - 2021-02-19

v1.0.0-alpha6 - 2021-02-17


  • Removal of "feedback" in rewards.
  • Changes to allow for TLS communication.

v1.0.0-alpha5 - 2021-01-28


  • Manage errors and exceptions at end of trial
  • Debug logging at critical points
  • Add access to the complete actor configuration from prehook implementations


  • Fix the instanciation of prehooks
  • Fix the environment "Final" flag not being set in some cases

v1.0.0-alpha4 - 2021-01-26


  • Catching exceptions that normally occur at the end of trials


  • Fix a crash occuring at the end of a trial in service actor implementations involving a KeyError.
  • tick_id was not being updated

v1.0.0-alpha3 - 2021-01-11


  • Create better errors around handling of implementation names


  • Support messsages
  • Avoid silent crashes by catching and logging exceptions thrown in async task and user code
  • Cancel tasks properly when they raise exceptions
  • Debug logging in strategic places


  • Send actions list to environment (as documented)
  • Store actor class name (string) for user facing session value and hide internal data
  • Client servicer code
  • Fix prometheus server

v1.0.0-alpha2 - 2020-12-09


  • Add missing cogment protobuf api files to the generated package.

v1.0.0-alpha1 - 2020-12-07

  • Initial alpha release, expect some breaking changes.

Project details

Release history Release notifications | RSS feed

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

cogment-2.10.1.tar.gz (101.3 kB view hashes)

Uploaded Source

Built Distribution

cogment-2.10.1-py3-none-any.whl (128.2 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page