ollama/examples/python-simplegenerate/README.md
Matt Williams 5a85070c22
Update readmes, requirements, packagejsons, etc for all examples (#1452)
Most of the examples needed updates of Readmes to show how to run them. Some of the requirements.txt files had extra content that wasn't needed, or missing altogether. Apparently some folks like to run npm start
to run typescript, so a script was added to all typescript examples which
hadn't been done before.

Basically just a lot of cleanup.

Signed-off-by: Matt Williams <m@technovangelist.com>
2023-12-22 09:10:41 -08:00

935 B

Simple Generate Example

This is a simple example using the Generate endpoint.

Running the Example

  1. Ensure you have the stablelm-zephyr model installed:

    ollama pull stablelm-zephyr
    
  2. Install the Python Requirements.

    pip install -r requirements.txt
    
  3. Run the example:

    python client.py
    

Review the Code

The main function simply asks for input, then passes that to the generate function. The output from generate is then passed back to generate on the next run.

The generate function uses requests.post to call /api/generate, passing the model, prompt, and context. The generate endpoint returns a stream of JSON blobs that are then iterated through, looking for the response values. That is then printed out. The final JSON object includes the full context of the conversation so far, and that is the return value from the function.