github/gpt-pilot
1
1
Fork 0
mirror of https://github.com/Pythagora-io/gpt-pilot.git synced 2026-01-09 21:27:53 -05:00
Code Issues Packages Projects Releases Wiki Activity

add jsonref dep

Browse Source
This commit is contained in:
Senko Rasic
2024-06-27 18:24:20 +02:00
parent 7bda7f1786
commit e4cfc00b1e
3 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
core/agents/convo.py
View File

@@ -97,6 +97,8 @@ class AgentConvo(Convo):
else: else:
return d return d
# We want to make the schema as simple as possible to avoid confusing the LLM,
# so we remove (dereference) all the refs we can and show the "final" schema version.
schema_txt = json.dumps(remove_defs(jsonref.loads(json.dumps(model.model_json_schema())))) schema_txt = json.dumps(remove_defs(jsonref.loads(json.dumps(model.model_json_schema()))))
self.user( self.user(
f"IMPORTANT: Your response MUST conform to this JSON schema:\n```\n{schema_txt}\n```." f"IMPORTANT: Your response MUST conform to this JSON schema:\n```\n{schema_txt}\n```."

1
pyproject.toml
View File

@@ -38,6 +38,7 @@ httpx = "^0.27.0"
alembic = "^1.13.1" alembic = "^1.13.1"
python-dotenv = "^1.0.1" python-dotenv = "^1.0.1"
prompt-toolkit = "^3.0.45" prompt-toolkit = "^3.0.45"
jsonref = "^1.1.0"
[tool.poetry.group.dev.dependencies] [tool.poetry.group.dev.dependencies]
pytest = "^8.1.1" pytest = "^8.1.1"

1
requirements.txt
View File

@@ -18,6 +18,7 @@ httpx==0.27.0
huggingface-hub==0.23.2 huggingface-hub==0.23.2
idna==3.7 idna==3.7
jinja2==3.1.4 jinja2==3.1.4
jsonref==1.1.0
mako==1.3.5 mako==1.3.5
markupsafe==2.1.5 markupsafe==2.1.5
openai==1.31.0 openai==1.31.0