📝 Update .gitignore to include .DS_Store and add untracked local-api-test directory
This commit is contained in:
1
.gitignore
vendored
1
.gitignore
vendored
@@ -1,2 +1,3 @@
|
|||||||
|
.DS_Store
|
||||||
.idea/
|
.idea/
|
||||||
.venv/
|
.venv/
|
||||||
|
|||||||
22
projects/local-api-test/main.py
Normal file
22
projects/local-api-test/main.py
Normal file
@@ -0,0 +1,22 @@
|
|||||||
|
import os
|
||||||
|
|
||||||
|
from openai import OpenAI
|
||||||
|
from openai._types import Timeout
|
||||||
|
|
||||||
|
|
||||||
|
def main():
|
||||||
|
client = OpenAI(
|
||||||
|
api_key=os.environ.get("OPENAI_API_KEY", "sk-dummy"),
|
||||||
|
base_url=os.environ.get("OPENAI_API_BASE", "http://localhost:11434/v1"),
|
||||||
|
timeout=Timeout(timeout=30),
|
||||||
|
)
|
||||||
|
|
||||||
|
response = client.chat.completions.create(
|
||||||
|
model=os.environ.get("OPENAI_MODEL", "llama3"),
|
||||||
|
messages=[{"role": "user", "content": "Hello!"}],
|
||||||
|
)
|
||||||
|
print(response.choices[0].message.content)
|
||||||
|
|
||||||
|
|
||||||
|
if __name__ == "__main__":
|
||||||
|
main()
|
||||||
8
projects/local-api-test/pyproject.toml
Normal file
8
projects/local-api-test/pyproject.toml
Normal file
@@ -0,0 +1,8 @@
|
|||||||
|
[project]
|
||||||
|
name = "local-ai-test"
|
||||||
|
version = "0.1.0"
|
||||||
|
dependencies = ["openai"]
|
||||||
|
|
||||||
|
[project.scripts]
|
||||||
|
run = "main:main"
|
||||||
|
|
||||||
Reference in New Issue
Block a user