diff --git a/README.md b/README.md index c9441685..3ce706e6 100644 --- a/README.md +++ b/README.md @@ -50,7 +50,9 @@ api_key = os.environ.get("OPENAI_API_KEY") # Initialize the language model llm = OpenAIChat( temperature=0.5, + model_name="gpt-4", openai_api_key=api_key, + max_tokens=4000 ) @@ -86,9 +88,10 @@ api_key = os.getenv("OPENAI_API_KEY") # Initialize the language agent llm = OpenAIChat( - openai_api_key=api_key, temperature=0.5, - max_tokens=3000, + model_name="gpt-4", + openai_api_key=api_key, + max_tokens=4000 ) diff --git a/example.py b/example.py index 9fc89485..6ed2cbab 100644 --- a/example.py +++ b/example.py @@ -15,6 +15,7 @@ api_key = os.environ.get("OPENAI_API_KEY") # Initialize the language model llm = OpenAIChat( temperature=0.5, + model_name="gpt-4", openai_api_key=api_key, ) diff --git a/sequential_workflow_example.py b/sequential_workflow_example.py index 1742f49c..38cf5559 100644 --- a/sequential_workflow_example.py +++ b/sequential_workflow_example.py @@ -11,10 +11,12 @@ api_key = os.getenv("OPENAI_API_KEY") # Initialize the language agent +# Initialize the language model llm = OpenAIChat( - openai_api_key=api_key, temperature=0.5, - max_tokens=2000, + model_name="gpt-4", + openai_api_key=api_key, + max_tokens=4000 )