fix hidden states and scores not being returned #47

	name: Python Lint

	on:
	push:
	branches: [ main ]
	pull_request:
	branches: [ main ]

	jobs:
	build:

	runs-on: ubuntu-latest
	strategy:
	matrix:
	python-version: [3.7, 3.8]

	steps:
	- uses: actions/checkout@v2
	- name: Set up Python ${{ matrix.python-version }}
	uses: actions/setup-python@v2
	with:
	python-version: ${{ matrix.python-version }}
	- name: Install dependencies
	run: \|
	python -m pip install --upgrade pip
	pip install -r requirements.txt
	pip install flake8 black mypy types-requests
	- name: Lint with Black
	run: \|
	# check if black would reformat anything
	black llm_client/ --check
	- name: Lint with flake8
	run: \|
	# stop the build if there are Python syntax errors or undefined names
	flake8 llm_client/ --count --select=C,E,F,W,B,B950 --ignore=E203,E501,E731,W503 --show-source --statistics
	# exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
	flake8 llm_client --count --exit-zero --max-complexity=10 --max-line-length=88 --statistics
	- name: Type Checking with MyPy
	run: \|
	# stop the build if there are type errors
	mypy --strict llm_client/

Provide feedback