File tree
Expand file treeCollapse file tree2 files changed
+23
-1
lines changed Expand file treeCollapse file tree2 files changed
+23
-1
lines changed Original file line number | Diff line number | Diff line change |
---|
@@ -2,4 +2,6 @@ fastapi==0.115.12
|
2 | 2 | uvicorn[standard]
|
3 | 3 | requests
|
4 | 4 | sqlmodel
|
5 |
| -psycopg[binary] |
| 5 | +psycopg[binary] |
| 6 | +langchain |
| 7 | +langchain-openai |
Original file line number | Diff line number | Diff line change |
---|
|
| 1 | +import os |
| 2 | + |
| 3 | +from langchain_openai import ChatOpenAI |
| 4 | + |
| 5 | + |
| 6 | +OPENAI_BASE_URL = os.environ.get('OPENAI_BASE_URL') or None |
| 7 | +OPENAI_MODEL_NAME = os.environ.get('OPENAI_MODEL_NAME') or 'gpt-4o-mini' |
| 8 | +OPENAI_API_KEY = os.environ.get("OPENAI_API_KEY") |
| 9 | +if not OPENAI_API_KEY: |
| 10 | +raise NotImplementedError("`OPENAI_API_KEY` is required") |
| 11 | + |
| 12 | + |
| 13 | +openai_params = { |
| 14 | +"model": OPENAI_MODEL_NAME, |
| 15 | +"api_key": OPENAI_API_KEY |
| 16 | +} |
| 17 | +if OPENAI_BASE_URL: |
| 18 | +openai_params['base_url'] = OPENAI_BASE_URL |
| 19 | + |
| 20 | +llm_base = ChatOpenAI(**openai_params) |
You can’t perform that action at this time.
0 commit comments