mirror of
https://github.com/aljazceru/gpt-engineer.git
synced 2025-12-17 04:35:17 +01:00
19a4c10b6e
* Added LangChain integration * Fixed issue created by git checkin process * Added ':' to characters to remove from end of file path * Tested initial migration to LangChain, removed comments and logging used for debugging * Tested initial migration to LangChain, removed comments and logging used for debugging * Converted camelCase to snake_case * Turns out we need the exception handling * Testing Hugging Face Integrations via LangChain * Added LangChain loadable models * Renames "qa" prompt to "clarify", since it's used in the "clarify" step, asking for clarification * Fixed loading model yaml files * Fixed streaming * Added modeldir cli option * Fixed typing * Fixed interaction with token logging * Fix spelling + dependency issues + typing * Fix spelling + tests * Removed unneeded logging which caused test to fail * Cleaned up code * Incorporated feedback - deleted unnecessary functions & logger.info - used LangChain ChatLLM instead of LLM to naturally communicate with gpt-4 - deleted loading model from yaml file, as LC doesn't offer this for ChatModels * Update gpt_engineer/steps.py Co-authored-by: Anton Osika <anton.osika@gmail.com> * Incorporated feedback - Fixed failing test - Removed parsing complexity by using # type: ignore - Replace every ocurence of ai.last_message_content with its content * Fixed test * Update gpt_engineer/steps.py --------- Co-authored-by: H <holden.robbins@gmail.com> Co-authored-by: Anton Osika <anton.osika@gmail.com>
39 lines
747 B
Python
39 lines
747 B
Python
import json
|
|
import pathlib
|
|
|
|
from typing import Union
|
|
|
|
import typer
|
|
|
|
from gpt_engineer.ai import AI
|
|
from gpt_engineer.chat_to_files import to_files
|
|
|
|
app = typer.Typer()
|
|
|
|
|
|
@app.command()
|
|
def main(
|
|
messages_path: str,
|
|
out_path: Union[str, None] = None,
|
|
model: str = "gpt-4",
|
|
temperature: float = 0.1,
|
|
):
|
|
ai = AI(
|
|
model_name=model,
|
|
temperature=temperature,
|
|
)
|
|
|
|
with open(messages_path) as f:
|
|
messages = json.load(f)
|
|
|
|
messages = ai.next(messages, step_name="rerun")
|
|
|
|
if out_path:
|
|
to_files(messages[-1]["content"], out_path)
|
|
with open(pathlib.Path(out_path) / "all_output.txt", "w") as f:
|
|
json.dump(messages[-1]["content"], f)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
app()
|