
    E{i                     J    d dl Zd dlmc mZ d dlmZ d dlm	Z	 d Z
d Zd ZdS )    N)Sessionc                  X   t          dd          } |                                 }|d         }d}||k    }|slt          j        d|fd||f          t          j        |          t          j        |          dz  }d	d
|iz  }t          t          j        |                    d x}x}}d S )Nzqwen/qwen3.6-plus:freeHIGH)modelreasoning_effortr   high==)z%(py1)s == %(py4)s)py1py4zassert %(py6)spy6)r   compact
@pytest_ar_call_reprcompare	_safereprAssertionError_format_explanation)sessionr   @py_assert0@py_assert3@py_assert2@py_format5@py_format7s          9/home/ubuntu/hermes-webui/tests/test_session_reasoning.py.test_session_compact_includes_reasoning_effortr      s    4vNNNGooG%&0&0&&0000000000&&00000&0000&00000000000000000000000000000    c                     t                      } | j        }d}||k    }|st          j        d|fd||f          dt	          j                    v st          j        |           rt          j        |           ndt          j        |          t          j        |          dz  }dd|iz  }t          t          j	        |                    d x}x}}d S )N r	   )z8%(py2)s
{%(py2)s = %(py0)s.reasoning_effort
} == %(py5)sr   )py0py2py5zassert %(py7)spy7)
r   r   r   r   @py_builtinslocals_should_repr_global_namer   r   r   )r   @py_assert1@py_assert4r   @py_format6@py_format8s         r   5test_session_stores_blank_reasoning_effort_by_defaultr*      s    iiG#)r)#r))))))))))#r))))))))))))7)))))7))))))#))))r)))))))))))))))))))))))))))))r   c            	         t           j        } d}d} | ||          }d}||k    }|st          j        d|fd||f          dt	          j                    v st          j        t                     rt          j        t                     ndt          j        |           t          j        |          t          j        |          t          j        |          t          j        |          dz  }dd|iz  }t          t          j	        |                    d x} x}x}x}x}}t           j        } d}d	} | ||          }d
}||k    }|st          j        d|fd||f          dt	          j                    v st          j        t                     rt          j        t                     ndt          j        |           t          j        |          t          j        |          t          j        |          t          j        |          dz  }dd|iz  }t          t          j	        |                    d x} x}x}x}x}}t           j        } d}d} | ||          }d
}||k    }|st          j        d|fd||f          dt	          j                    v st          j        t                     rt          j        t                     ndt          j        |           t          j        |          t          j        |          t          j        |          t          j        |          dz  }dd|iz  }t          t          j	        |                    d x} x}x}x}x}}d S )Nz@openai-codex:gpt-5.4mediumr	   )zf%(py8)s
{%(py8)s = %(py2)s
{%(py2)s = %(py0)s.clamp_reasoning_effort
}(%(py4)s, %(py6)s)
} == %(py11)sconfig)r   r    r   r   py8py11zassert %(py13)spy13xhighr   z@deepseek:deepseek-reasonerr   )
r-   clamp_reasoning_effortr   r   r#   r$   r%   r   r   r   )r&   r   @py_assert5@py_assert7@py_assert10@py_assert9@py_format12@py_format14s           r   6test_clamp_reasoning_effort_rejects_unsupported_valuesr9      s   (W)@W(W(()@(KKWxWKxWWWWWWWWWWKxWWWWWWWWWWWW6WWWWW6WWWWWW(WWWW)@WWWW(WWWWKWWWWxWWWWWWWWWWWWWWWWWWWWWWWWWWWWWWWWW(P)@P'P(()@'JJPbPJbPPPPPPPPPPJbPPPPPPPPPPPP6PPPPP6PPPPPP(PPPP)@PPPP'PPPPJPPPPbPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPP(U)FUU(()FOOUSUUOSUUUUUUUUUUUOSUUUUUUUUUUUUU6UUUUU6UUUUUU(UUUU)FUUUUUUUUOUUUUSUUUUUUUUUUUUUUUUUUUUUUUUUUUUUUUUUUUUr   )builtinsr#   _pytest.assertion.rewrite	assertionrewriter   
api.configr-   
api.modelsr   r   r*   r9    r   r   <module>rA      s                             1 1 1* * *
V V V V Vr   