alephant-python-sdk/.github/workflows/ci.yml at main · AlephantAI/alephant-python-sdk · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
name: CI

on:
  push:
    branches: ["main"]
  pull_request:

jobs:
  tests:
    runs-on: ubuntu-latest
    strategy:
      fail-fast: false
      matrix:
        python-version: ["3.10", "3.12"]
    steps:
      - uses: actions/checkout@v4
      - uses: actions/setup-python@v5
        with:
          python-version: ${{ matrix.python-version }}
      - name: Install dev dependencies
        run: python -m pip install -e "packages/alephantai[dev]" -e "packages/langchain-alephantai[dev]"
      - name: Run tests
        run: pytest -v
      - name: Run ruff
        run: ruff check .

  optional-extras:
    runs-on: ubuntu-latest
    strategy:
      fail-fast: false
      matrix:
        extra: ["langchain", "llamaindex"]
    steps:
      - uses: actions/checkout@v4
      - uses: actions/setup-python@v5
        with:
          python-version: "3.12"
      - name: Install optional extra
        run: python -m pip install -e "packages/alephantai[${{ matrix.extra }}]"
      - name: Check dependencies
        run: python -m pip check
      - name: Smoke LangChain constructors
        if: matrix.extra == 'langchain'
        run: |
          python - <<'PY'
          from alephantai import AlephantGatewayContext
          from alephantai.langchain import create_chat_openai

          llm = create_chat_openai(
              api_key="vk-test",
              model="gpt-4o-mini",
              context=AlephantGatewayContext(session_id="sess-ci"),
          )
          assert type(llm).__name__ == "ChatOpenAI"
          PY
      - name: Smoke LlamaIndex constructors
        if: matrix.extra == 'llamaindex'
        run: |
          python - <<'PY'
          from alephantai import AlephantGatewayContext
          from alephantai.llamaindex import create_openai_embedding, create_openai_llm

          ctx = AlephantGatewayContext(session_id="sess-ci")
          llm = create_openai_llm(api_key="vk-test", model="gpt-4o-mini", context=ctx)
          embedding = create_openai_embedding(
              api_key="vk-test",
              model="text-embedding-3-small",
              context=ctx,
          )
          assert type(llm).__name__ == "OpenAI"
          assert type(embedding).__name__ == "OpenAIEmbedding"
          PY

  langchain-package:
    runs-on: ubuntu-latest
    steps:
      - uses: actions/checkout@v4
      - uses: actions/setup-python@v5
        with:
          python-version: "3.12"
      - name: Install package with standard test dependencies
        run: python -m pip install -e "packages/alephantai" -e "packages/langchain-alephantai[dev]"
      - name: Run LangChain package tests
        run: pytest packages/langchain-alephantai/tests -m "not integration" -v
      - name: Smoke ChatAlephantAI constructor
        run: |
          python - <<'PY'
          from langchain_alephantai import ChatAlephantAI

          llm = ChatAlephantAI(
              api_key="vk-test",
              model="gpt-4o-mini",
              session_id="sess-ci",
          )
          assert type(llm).__name__ == "ChatAlephantAI"
          assert llm.default_headers["Alephant-Session-Id"] == "sess-ci"
          PY